[MRG] preserve column and branch order from TreeModel definition