add test for remove