diff --git a/TaskB01/run b/TaskB01/run
index 4ef7991..ada0cda 100644
--- a/TaskB01/run
+++ b/TaskB01/run
@@ -45,11 +45,11 @@ class automata:
     def check_if_accepted(self):
         return not set(self.state).isdisjoint(self.accepting_states)
 
-    def get_node_transition(self, query):
-        for i in self.storage:
+    def get_node_transition(self, q, i):
+        self.storage[q]
             if i[0] == query:
                 return i[1]
-        return []          
+        return []
 
 
 auto = automata()
@@ -59,7 +59,7 @@ for line in sys.stdin:
 
 print(auto)
 
-f = open(sys.argv[1], 'r')
+# f = open(sys.argv[1], 'r')
 
-for line in f:
-    print(auto.test_string(line))
+# for line in f:
+#     print(auto.test_string(line))