More tree gen stuff