Add edit scenarios

It should work now