Coursera - Data Science Course assignment