add tests for ModelBuilder class