@charmcli everything went pretty smooth with updating my custom tree model:

https://github.com/mariusor/bubbles-tree/pull/4

I left some impressions in the PR description.