Data Scientist Course Homework