minor change to balanced tree example