Assignment for Data Science specialization from Coursera