Code highlighting tweaks