Update ControllerGenerator.php