Task 3: fixing the end times in the scheduler