Modify controller.php