Assignment for Data Science Toolbox class