Test method added for the remove method