0.3 Add the edit feature