Created as part of the assigment from "The Data Scientist Toolbox"