diff --git a/bin/submit.py b/bin/submit.py
index 92bdb6f..e43013f 100644
--- a/bin/submit.py
+++ b/bin/submit.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python
 
+import sys
+
 from multiprocessing.managers import BaseManager
 
 import numpy as np
@@ -13,8 +15,10 @@ m = QueueManager(address=('gannon', 50000), authkey='asdf')
 m.connect()
 
 if __name__ == '__main__':
+  count = int(sys.argv[1])
+
   inq = m.get_inqueue()
 
-  for i in xrange(1000):
+  for i in xrange(count):
     X = np.random.random((1,3))[0]
     inq.put((i,X))