minor changes to graph