Much better plots

MAny improvements made to the plots in the notebook.