More work on tree balancing