Update the main controller for the replay button