Add optional support for highlight.js