Going to use networkx for the example