Added restore feature.