Assignment for Data Science Toolbox course project