Improve style of the comments