Add another test case for remove()