add return for tree widget