Added tests for Collection, resolves #4