Minor Changes -> Also Changed Vis. Part. We can accept a Tree from Modeling now :)