Part of the course project for "The Data Scientist's Toolbox" course