Add a functional remove method