Update GenerateController.php