Problem 5 correct solve