Data Science course assignment1