Updated the example notebook