Data Science Course assignment