Re-introduce the 'end_force' function as a task in order for the cooling function to be called at the right place in the time-step.