Data science toolbox task no 2