Added Controller class