Adding Graph Viz correction