Fix missing probabilities

main.py

@@ -22,7 +22,7 @@ def add_dict(a, b):
     for key in b:
         if key not in a:
             temp[key] = b[key]
-    return a
+    return temp
 
 class Chatbot:
     def __init__(self, name=None, letter_replace: bool = True, data: dict = None, frequency_weight: float = 0, div_by_len: bool = False):