Add edit function

Edit function