Second part of the course project for The Data Scientists Toolbox