Adding the edit feature continuation