Redone tree imbalance plots after fixing bug