Made a model helper that is the storage. It's working but not finished. Just realised i could do this eager with a state flag on each tree.