2d graphs in python