Implemented snapshot restore functionality