Part of the assignment for The Data Scientist's Toolbox classes at John Hopkins and Coursera