udpated example notebook