fixed fig.8 and re organizing the notebook