1st assignment from Data Scientist Toolbox