minor tweaks to the generator example