Major updates to the python notebooks

This follows on from !137 and !136, integrating the other changes
that had been made in the git repo and trying to further tidy up.

Author: Zach Marshall