Minor fixes to code highlighting
Minor fixes to code highlighting