Fixing TreeNode and view. Does not work yet.