For the project for the data scientist toolkit class