Data Science Coursera Class assignment