With Shuffle before partitionPersist