Part of assignment week 3 of Data Science Toolbox