added click to highlight function on code element
added click to highlight function on code element