Coursera Data Science Class assignment