Add code block highlight