Assignment for data science course on coursera