refactored remove to actually take an object and not a query