Part of the course project for The Data Scientist’s Toolbox