Removed controller class as it is not needed