Small improvements to comments