Undid change to controller class