added a controller class