Data Scientist Assignment 1