Feature: Restore state