working edit feature