diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index a024c85..00f1e7e 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -1,5 +1,6 @@
 add_executable(markov
     ./main.cpp
+    ./generator.cpp
 )
 
 target_include_directories(markov PUBLIC ${CMAKE_CURRENT_LIST_DIR})
diff --git a/src/generator.cpp b/src/generator.cpp
new file mode 100644
index 0000000..e69de29
diff --git a/src/generator.hpp b/src/generator.hpp
new file mode 100644
index 0000000..b4d95bd
--- /dev/null
+++ b/src/generator.hpp
@@ -0,0 +1,13 @@
+#ifndef GENERATOR_HPP
+#define GENERATOR_HPP
+
+struct Generator {
+public:
+    Generator(void);
+    void SetArguments(int argc, char* argv[]);
+private:
+    int prefixLength = 0;
+    int outputLength = 0;
+};
+
+#endif // !GENERATOR_HPP
diff --git a/src/main.cpp b/src/main.cpp
index 29b2fa0..aba9d35 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,6 +1,7 @@
-#include <iostream>
+#include "generator.hpp"
 
-int main(void) {
-    std::cout << "Hello world" << std::endl;
+int main(int argc, char* argv[]) {
+    Generator markovChain;
+    markovChain.SetArguments(argc, argv);
     return 0;
 }