Update code highlighting