Update the Controller class