Add missing controller class