Add edit tests