add test for method remove