WIP - Support index samples in counts tracker

Todo:
- Merger
- Tests