This is a part of The Data Scientist's Toolbox course project