Updated notebook demo