Somehow got the drag to work with immutable tree. But with some bugs