Modify treetable script