I'd suggest this simple performant pattern for:

```python
from itertools import chain

group_stats = dict(chain(original_group_stats.iteritems(), group_stats.iteritems())
```

Better yet, we could put this in a `lib/utils` helper method, I believe we're doing the same thing in other places in the code