Moved graph generation function into generate.py.