def modify_state_for_body(s, body):
            s.layers['segmentation'].segments = frozenset([body.segment_id])
            s.voxel_coordinates = body.bbox_start + body.bbox_size // 2
def insert(root, key):
    left, right = split(root, key)
    return merge(merge(left, Treap(key)), right)