added the edit feature