diff --git a/Python Code/SoS_Acquisition.py b/Python Code/SoS_Acquisition.py
index 9335131..b90152a 100644
--- a/Python Code/SoS_Acquisition.py	
+++ b/Python Code/SoS_Acquisition.py	
@@ -12,6 +12,10 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import time
+import os
+# Create directories if they don't exist
+os.makedirs("data/X/", exist_ok=True)
+os.makedirs("data/Y/", exist_ok=True
 
 
 '''