added more tests
refactoring + added more tests for collection:remove