Also implement support for Python generators