add drop_column kwarg