Assignment for Coursera Data Science Toolbox class