Assignment for data scientist toolbox course on Coursera