Created for the course project of Data Scientist's Toolbox course