Update ModelBuilder.cs