Updated notebook to use recent syntax changes