Added logic to end a shift.