added controller class