Tristan Stevens

Check out #11, it is not merged but they did fix quite some bugs in the notebook there.

Finding this lost PR solved quite some issues in making the code run for me. Maybe the authors can merge the PR.