working out tree widget bugs