Added a few remove() tests
Merge pull request #2 from dragonai/patch-1

Added a few remove() tests