added test for remove