adding controller class