Update ShuffleAndSort.java