Added unit test for Collection