Update IPython notebook.

Add code to make the extra figures.