Correction on problem 18