Minor fixes to plot_phase_diagram method.