Solved point 4 of assignment 2