GA's Data Science course homework