Playing around with the networkx module