Getting familiar with the networkX library