Slight improvement in the graph drawing method of the python module