Add optional 'key' argument to $remove