Update code block highlight