Fixing a few minor issues in the graph generation.