Add missing highlighting