Tweaked gen_graphs.py