write tests for remove