now also works in the notebook() as described in the README, added analytic comparison and helper funcs for plots