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