Modify the controller