Add yield function