Update code highlight