Python Generate A Graph