From 9ae148e5ee589e32653112128759a43cb6e8f039 Mon Sep 17 00:00:00 2001
From: Trianta <56975502+Trimutex@users.noreply.github.com>
Date: Thu, 7 Dec 2023 23:58:54 -0600
Subject: [PATCH] Minor code cleanup before continuing

---
 src/generator.cpp | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/src/generator.cpp b/src/generator.cpp
index 3f32cdb..d9edafa 100644
--- a/src/generator.cpp
+++ b/src/generator.cpp
@@ -3,22 +3,19 @@
 
 void Generator::SetArguments(int argc, char* argv[]) {
     std::string tempStr;
-    for (int i = 1; i < argc; ++i) {
+    for (int i = 1; i < argc; i += 2) {
         tempStr.assign(argv[i]);
         if (tempStr == "-i") {
             setup.isFileSet = true;
-            ++i;
-            setup.filename.assign(argv[i]);
+            setup.filename.assign(argv[i+1]);
         }
         if (tempStr == "-k") {
             setup.isPrefixSet = true;
-            ++i;
-            setup.prefixLength = std::stoi(argv[i]);
+            setup.prefixLength = std::stoi(argv[i+1]);
         }
         if (tempStr == "-n") {
             setup.isOutputSet = true;
-            ++i;
-            setup.outputLength = std::stoi(argv[i]);
+            setup.outputLength = std::stoi(argv[i+1]);
         }
         if (tempStr == "-h") {
             PrintUsage();