Add missing UI component