You can see the approach you describe detailed in this notebook. Note that this notebook has yet to be updated to reflect v4, so I’m not 100% that it will work exactly as illustrated in that notebook, but hopefully it gets you in the right direction. Feel free to here if something seems off.