def update_priority(self, tree_idx_list, priority_list):
        """ Update priorities of the elements in the tree """
        for tree_idx, priority, segment_tree in zip(tree_idx_list, priority_list, self.segment_trees):
            segment_tree.update(tree_idx, priority)