Updated example notebook