Any chance you could push an end-to-end example as a jupyter notebook? It's really hard to follow the codebase... thanks!