Work also with newer networkx (tested with 2.1).

Created by: he32

Reportedly still works with old version. This a suggested fix for isue #1 (closed).