Add support for syntax highlighting