Separate the controller class