Add highlighting to code blocks