Add edit functionality