Highlight code types and fix parsing of code sections.