Data Science coursera assignment