diff --git a/okxtrading2.0.py b/okxtrading2.0.py
index cc66471..03c18cd 100644
--- a/okxtrading2.0.py
+++ b/okxtrading2.0.py
@@ -3,6 +3,7 @@ import time
 import logging
 import pandas as pd
 import numpy as np
+import signal
 import requests
 import json
 import hashlib
@@ -111,7 +112,7 @@ def main():
         runner.start_all_strategies()
         while True:
             time.sleep(5)
-            
+
     # Add debug mode switch
     debug_mode = False