Part of The Data Scientist's Toolbox course project