1 files changed, 19 insertions, 1 deletions
diff --git a/src/game/edit.cpp b/src/game/edit.cpp
index 53605ca..b2d2542 100644
--- a/src/game/edit.cpp
+++ b/src/game/edit.cpp
@@ -1,9 +1,27 @@
 #include "Game.hpp"
+#include "../utils/Utils.hpp"
 
 #include <cstdlib>
 #include <cstdio>
 #include <iostream>
+#include <string>
+
+int Game::editDiff(){
+    Utils util;
 
-void Game::editDiff(){
     std::cout << "Now editing custom difficulty" << std::endl;
+
+    std::cout << "Board width: ";
+    util.catchInputInt(Game::boardWidth);
+    if(util.catchReturn(util.catchInputInt(Game::boardWidth)) != 0) return 1;
+
+    std::cout << "Board height: ";
+    util.catchInputInt(Game::boardHeight);
+    if(util.catchReturn(util.catchInputInt(Game::boardHeight)) != 0) return 1;
+
+    std::cout << "Mines: ";
+    util.catchInputInt(Game::mines);
+    if(util.catchReturn(util.catchInputInt(Game::mines)) != 0) return 1;
+
+    return 0;
 }
 \ No newline at end of file