add highlight function