Data Science course assignment task