optimize highlight function