Add edit back in