CPS Transformer: Add support for yield expressions in while