minor changes as I adjust the plotting
Done tweaking the best plot