Fixing highlighting styles