updated files with correct code