Added test method for remove()