Fixed some comments for problem 21