Improve: Memory footprint and performance of gather_stats