Refactor collect_data and group_data

**Is your feature request related to a problem? Please describe.**
Refactor as one for enhancement

**Describe the solution you'd like**
One method ``gather_results`` that can handle both condition: 1) with idx, 2) without idx

**Describe alternatives you've considered**

**Additional context**
