Controller class added