added edit feature