Update notebook to use a new simpler load function