Add some functionality to save state manager.