refactor, add spark.shuffle.sort=False