add code highlight function