CHANGES:

1. `thin_samples['init_state']` --> `thin_samples['initial_state']`
2. `thin_samples['events']` --> `thin_samples['seir']`
3. add time dimension to next generation matrix
4. update downstream scripts to changes.