Modify interface of structure/segment_tree