Starting handling game loop

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main.cpp b/src/main.cpp
index 88b8ffa..ef12c62 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -4,6 +4,7 @@ using namespace std;
 
 int main(){
 	Game game;
+	bool inGame = false;
 	int choice;
 
 	cout << "Welcome to minesweeper 0.1.0\n";
@@ -18,12 +19,21 @@ int main(){
 
 	if(choice == 1){
 		game.initDiff(0);
+		inGame = true;
 	} else if(choice == 2){
 		game.initDiff(1);
+		inGame = true;
 	} else if(choice == 3){
 		game.initDiff(2);
+		inGame = true;
 	} else if(choice == 4){
 		game.editDiff();
+		inGame = true;
+	}
+
+	while(inGame){
+		game.initBoard();
+		game.displayBoard();
 	}
 
 	return 0;