Add edit icon to UI