diff --git a/code/TSP_solver.py b/code/TSP_solver.py
index 2322776..d7028b6 100644
--- a/code/TSP_solver.py
+++ b/code/TSP_solver.py
@@ -32,7 +32,7 @@ class Solver_TSP:
         self.solved = True
         return self.solution
 
-    def nn(self, instance_):
+    def nn(self, instance_, starting_node=0):
         pass
 
     def plot_solution(self):
diff --git a/code/__init__.py b/code/__init__.py
index b7abdfa..80c4d39 100644
--- a/code/__init__.py
+++ b/code/__init__.py
@@ -1,2 +1,8 @@
-from code.TSP_solver import *
-from code.io_tsp import *
+import os
+
+if 'AI' in os.getcwd():
+    from code.TSP_solver import *
+    from code.io_tsp import *
+else:
+    from AI2019.code.TSP_solver import *
+    from AI2019.code.io_tsp import *
\ No newline at end of file