Updating ModelGenerator