Minor change to python graphing script.