Adds state restoration for non-initial controller via restoration class