For 3rd assignment of Data Scientist toolbox course