Added edit button when they are editting