Update TreeView and UI