Data Scientist assignment