added a bit more to the tree creation, need to work on it more, but  for now it should be fine