Improved highlighting in translation mode