Improved the graph visualization