Update ModelBuilder.php