Add test for mapRemove