First homework assignment for Data Science