Added problem 2 and 3