cleaned up ipython notebook to make the plots the way they currently are in paper draft