Add ModelBuilder class