Correction problem 4