Problem 4 completed