modify edit method