add hljs class to the highlighted elements