Add problem 2 to 7