Add comment explaining collectPartitions's use