docs and ipynb tweaks