Address reviewer's suggestions on distributed partitioning