diff --git a/src/Main.hs b/src/Main.hs
index 280c8d6..1639f01 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -12,9 +12,11 @@ mkPop = population 100 (I prios [])
 
 main :: IO ()
 main = do
+  args <- getArgs
+  let t = fromMaybe 100 $ headMay args >>= readMaybe
   hSetBuffering stdout NoBuffering
   pop <- mkPop
-  pop' <- runEffect $ for (run 2 1 pop (steps 100)) log
+  pop' <- runEffect $ for (run 2 1 pop (steps t)) log
   res <- bests 5 pop'
   sequence_ $ format <$> res
   where