Adding Py/yield