I just realized this is exactly the example provided in the repo which I haven't run yet! Thanks for adding this!