adding the edit feature