Implement remove_key