diff --git a/neural_network.py b/neural_network.py
index d3c525d..e3bc874 100644
--- a/neural_network.py
+++ b/neural_network.py
@@ -152,11 +152,11 @@ for i in range(29):
 # Used for testing data
 
 for i in range(11, 29):
-    filename = join('/pub/faculty_share/daugher/datafiles/data/', str(i) + '.bin')
+    filename = join('/pub/faculty_share/daugher/datafiles/data/', str(i) + 'states.bin')
 
     print(i)
 
-    with open('/pub/faculty_share/daugher/datafiles/data/18states.bin', 'rb') as f:
+    with open(filename, 'rb') as f:
 
         for i in range(2000):
             data = f.read(8)
@@ -169,7 +169,7 @@ for i in range(11, 29):
 
         testing_target = []
 
-        while(data):
+        while(data and counter < 10000):
             bin_data = reduce(format_input, list(data), [])
             bin_data.reverse()
             bin_data = bin_data[16:]