Minor change to edit button