Add `generation_time` to output