Add test for removing values