Major update of simple pyroot notebook

This gets things working again on both Binder and Colab

Author: Zach Marshall