Highlight code in comments