Used for assignment 1 of the Data Scientist's Toolbox Coursera course