Minor improvement to comments