Homework for data science course