Added missing TreeModel class