updated code to use hljs