Added graph vizualization