Update draw figure script. Adjust output a bit.