fix unittest, update notebook
documentation