Assignment for Data Science course