For Data Scientist assignment