updated notebook for nbviewer preview