fix comments on problem 21