changed controller class