part of the assignment for data scientist's toolbox