data science, homework 1