add test for remove method