Improve comments of the code.