Shameless plug but made a similar tree here: https://sajarin.com/blog/modeltree/