better loop over tree (it's pyroot, bro)