Added edit func