Adding tests for collection