Add proof state persistance
Add proof state persistance