For the class project in The Data Scientist's Toolbox