Add controller class