add remove_keys option