Assignment 1 for Data Science Coursera