this is for the data scientist's toolbox assignment