Add problem 2 and 3