Controller.php updated