Update problem-5.py

Author: SohaibBazaz

problem-5/problem-5.py

@@ -1,17 +1,12 @@
-def findTheDifference():
-    s = input("Enter the first string: ")
-    t = input("Enter the second string: ")
-
-    for i in t:
-        if i not in s:
-            result = i
+def findTheDifference(s, t):
+    
+    for i in range(len(t)):
+        if s[i] != t[i]:
+            result = t[i]
             break
-        else:
-            if s.count(i) != t.count(i):
-                result = i
-                break
 
     print("The difference is:", result)
 
 def main():
-    findTheDifference()
+    findTheDifference("abcd", "abced")
+main()