Add TreeIter back to mod.rs