Highlight code snippets correctly