Add support for code highlighting