Added restore feature