add highlight function for code