updated files to improve edit function