Added test for .remove