update gen2.py (cosmetic)
update gen_mod.py (working maybe)