Add: Tests for remove method