For final project of "The Data Scientist's Toolbox"