Parameterized the treetable object better