Added problem 4