Got the edit feature working