making a graph in python