Coursera Introduction to Data Science Assignment 2