Update to the plot

Created by: scarlehoff

I needed two commits because having a slightly different version of matplotlib was slightly different. I'm really hoping this one last.