Add test for remove()