Add method to load state from store.