diff --git a/collect.py b/collect.py
index f6b336d..1c4ca24 100644
--- a/collect.py
+++ b/collect.py
@@ -6,7 +6,7 @@ import os
 import shutil
 import socket
 import platform
-import time
+import argparse
 
 from datetime import datetime
 
@@ -367,13 +367,34 @@ def is_two_digit_number(number):
 def main():
     welcome()
 
+    # setup argparse
+    parser = argparse.ArgumentParser(description="TrueNAS Performance Capture Script")
+    parser.add_argument(
+        "runMinutes", nargs="?", help="Number of minutes to collect data.", default=0
+    )
+    parser.add_argument(
+        "waitMinutes",
+        nargs="?",
+        help="Number of minutes to wait before collecting data.",
+        default=0,
+    )
+
+    args = parser.parse_args()
+    minutes = int(args.runMinutes)
+    minutesToWait = int(args.waitMinutes)
+
     try:
-        minutes = int(input("Enter the duration in minutes: "))
-        minutesToWait = int(input("Enter the delay before capture in minutes: "))
+        if not minutes:
+            minutes = int(input("Enter the duration in minutes: "))
+            minutesToWait = int(input("Enter the delay before capture in minutes: "))
 
         if not minutesToWait:
             minutesToWait = 0
 
+        print(
+            f"Set to wait {minutesToWait} minutes and then capture for {minutes} minutes."
+        )
+
         if minutesToWait:
             print(f"Delaying capture by {minutesToWait} minutes...")
             for i in range(minutesToWait):