diff --git a/twd_solver.py b/twd_solver.py
index 47377e4c72a137fabb4b56fd8ef7ae71b425aacb..add2d9cf805de09fc6ea2a8c66cf0c790d44e303 100644
--- a/twd_solver.py
+++ b/twd_solver.py
@@ -1,7 +1,11 @@
+import argparse
 import os
 import subprocess
 
-from solvers.twd import solver
+if __name__ == '__main__':
+    import solver
+else:
+    from solvers.twd import solver
 
 proc = None
 stop_flag = False
@@ -68,7 +72,6 @@ def solve(params):
         # print(errs)
 
     solution = ''
-    print(returncode)
     if returncode == 10:    # 10 when SAT
         status = 'done'
         if not stop_flag: