data scientist's toolbox - assign1