PR #139 introduce some API changes that break the example jupyter notebook. We should update it to work, using the new API