Improve code highlighting.