How to build a graph in Python