For "The Data Scientist's Toolbox" final project