Add problem 3 and 4