Add edit/update methods