improvements on graph viz stuff