Correct the solution to number 4