Problem 4 complete