updated notebook to use new library improvements