Small improvement to comments