Updated restore functionality