Fix select_columns columns order
Fix the order of the columns in dataset.features when the order changes with `dataset.select_columns()`.

I also fixed the same issue for `dataset.flatten()`

Close https://github.com/huggingface/datasets/issues/5993