Added graph visualization