Updates to example notebook