Tweak code highlighting