small mods to the generator..