Added basic networkx graph visualziation code