Data Scientist Course 1 Assignment