Assignment nr 3 for data science course