Adding model builder for collection loader