Edited treegen again