create a graph data structure in python