Retake/reset stage working.
Adding saving screen