python yield at work