Small improvements to the comments