https://raw.githubusercontent.com/ayushshivani/aicrowd_educational_baselines/master/SPCRT_baseline.ipynb