Part of the course project of Data Scientists Toolbox