Make simple loops yield automatically