Add yield function to reschedule the current job