Improve comments on exercise 2