Add changes for the scheduler in Save Assessment flow