Refactor
Use highlight.js directly