Add tests for collection