updated files to original code