diff --git a/PythonScripts/test_disk.py b/PythonScripts/test_disk.py
index ad03a6c..4a3c3a2 100644
--- a/PythonScripts/test_disk.py
+++ b/PythonScripts/test_disk.py
@@ -58,27 +58,16 @@ def get_device_info(device):
     output = subprocess.run(
         f"sudo smartctl -i {device}", shell=True, capture_output=True, text=True
     ).stdout
-
     model = serial = "unknown"
-
     for line in output.splitlines():
-        if any(x in line for x in ["Device Model", "Model Number", "Product"]):
+        if any(x in line for x in ["Device Model", "Model Family", "Model Number"]):
             model = line.split(":", 1)[-1].strip().replace(" ", "_")
-        elif "Vendor:" in line and model == "unknown":
-            vendor = line.split(":", 1)[-1].strip().replace(" ", "_")
-            model = vendor  # temporary, could be combined later
-        elif "Serial Number" in line or "Serial number" in line:
+        elif "Serial Number" in line:
             serial = line.split(":", 1)[-1].strip()
-        elif "Logical Unit id" in line and serial == "unknown":
-            serial = line.split()[-1].strip()
-
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-
     if model == "unknown" or serial == "unknown":
         print("Failed to extract model or serial. Using timestamp for filename.")
-        return f"drive_test_{timestamp}.log"
-
-    return f"{model}_{serial}_test_{timestamp}.log"
+        return f"drive_test_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
+    return f"{model}_{serial}_test.log"
 
 
 def show_smart_info(device, log_file):
@@ -121,7 +110,7 @@ def fill_random(device, log_file):
     print("Random data write complete.")
 
 
-def run_fio(device, rw_mode, log_file, bs="4k", runtime=60):
+def run_fio(device, rw_mode, log_file, bs="4k", runtime=30):
     print(f"Running fio test: {rw_mode}")
     cmd = (
         f"sudo fio --name={rw_mode}_test "
@@ -163,7 +152,7 @@ def main():
         run_fio(device, "read", log_file)
         run_fio(device, "randread", log_file)
         run_fio(device, "randwrite", log_file)
-        run_fio(device, "randread", log_file, bs="512", runtime=30)  # IOPS test
+        run_fio(device, "randread", log_file, bs="512", runtime=10)  # IOPS test
 
         print(f"\nTest completed. Results saved to {log_filename}")