Update model collection API.
Add tests.