Added the Controller class