Added restore capablity to UI.