Adds tests for `remove()`

See #212 and also #215.