Assignment for data science coursera