Add state saving to Tally