fixed imports in test tree
updated notebook for presentation