Updating my Tree Program