Improved highlighting rule for comments