Improved highlighting rule for import