ENH: Implemented TreeModel