Add missing cases