So I can tweak the figures
Nicer plot for paper
New figure