homework for data science course