Add ES6 yield support