Part of the assignment for the Data Science Toolbox course