TreeView User Set Expand / Collapse

Hi All,
Currently, When new treeview record added, edited and deleted, all treview nodes expands. Also when show form same behavior.
Is it possible to keep user set expand / collapse when adding etc treeview items. For instance, if I had collapsed some notes, I wanted it to stay collapsed that node until I change regardless adding / editing / deleting treview node/s and when showing the form

Please see the sample project below, which includes Dmitry's script for moving same level nodes and Derek's expand / collapse script.