Bring back restore functionality