Added problem 4 - without comments