homework for Data Science