Actually removed import networkx line. :)