Add edit controls