Update GraphViewer code