I worked on deleting the button since we want training modal to close automatically upon completion. I added the set variables into TrainingProgress().