this is to complete the assignment for Data Scientist Toolbox