diff --git a/AutoCompModule.py b/AutoCompModule.py
index 7870281..c59761c 100644
--- a/AutoCompModule.py
+++ b/AutoCompModule.py
@@ -1,12 +1,13 @@
 import pymongo, os, sys, re
 from pymongo import MongoClient
+from pymongo import Connection
 
 class AutoCompModule:
 
     def __init__(self,DBName):
-        self.dict = MongoClient().locals()[DBName].dict
-        self.dictBy2 = MongoClient().locals()[DBName].dictBy2
-        self.dictBy3 = MongoClient().locals()[DBName].dictBy3
+        self.dict = Connection()[DBName]['dict']
+        self.dictBy2 = Connection()[DBName]['dictBy2']
+        self.dictBy3 = Connection()[DBName]['dictBy3']
     
     def learnSingle(self,fileName):
         input = open(fileName, encoding='utf-8')
diff --git a/Step_1_Init.py b/Step_1_Init.py
index 15bd5f3..d8628b2 100644
--- a/Step_1_Init.py
+++ b/Step_1_Init.py
@@ -2,7 +2,7 @@
 from pymongo import MongoClient
 
 def main():
-    x=5
+    x=7;
     input = open(sys.argv[1], encoding='utf-8')
     words = MongoClient().db.words
     words2 = MongoClient().db.words2