Data Science course assignment 1