Add retake button and save button