Data Science course exercise