Update ModelGenerator.php