Remove update behavior, Make new_labels an int param