TreeB improve