updated notebook for new output formats