Add graph visualization support in pytools.visualization