Instead of shuffling it is probably possible to keep the `idx` for later.
And make `y = y[idx]`, we don't probably gain much with that.