homework for Coursera Data Scientist's Toolbox class