Edit Controller.php