Some changes to the script to produce the plot we use in scipy ( basically fix the subplot problem, and play a bit with the axis) '