Added unit test for remove.