I'm honored to have the great SimonW comment (lol) I've just pushed an update that incorporates a lot of the rendering from your notebook into the main repo - Thanks for the suggestion!