Should allow users to write sensible state upgrade methods

Right now unusable b/c upgrade method uses current config dataclass to load state... Of course that can't work when fields have been added / removed. Better approach needed.