graph representation in python