How to keep track of expanded state when updating data

[RamsayRomero]

I am attempting to add lazy loading to fetch child nodes dynamically when a node is expanded. I can update the new data but without keeping track of the expanded state for every node, the new data will be collapsed when loaded. Does anyone have any ideas of how I can keep the current expanded state of my tree while adding new nodes to the data?