Adding the Controller