Update highlighting for code blocks