Add container edit function