Data science tool box assignment part 2