Tweak code highlighting a bit