Better folded code highlighting