For the course project for Data Scientist's Toolbox