Improved comments, updated IPython notebook