Changed victory end to reset

src/main/java/Minesweeper/Cell.java

@@ -8,6 +8,7 @@ import java.awt.event.MouseAdapter;
 import java.awt.event.MouseEvent;
 
 import javax.swing.JButton;
+import javax.swing.JOptionPane;
 
 enum CellType {
 	Number, Bomb
@@ -53,6 +54,7 @@ public class Cell extends JButton {
 			if (type != CellType.Bomb) {
 				flood();
 			} else {
+				JOptionPane.showMessageDialog(getParent(),"KABOOM! Try again!");
 				playfield.reset();
 			}
 		}

src/main/java/Minesweeper/Playfield.java

@@ -12,8 +12,7 @@ public class Playfield {
 	public Cell[][] cells;
 	private int bombAmount;
 	private int cellsFlooded = 0;
-	
-	
+
 	public Playfield(MinesweeperGame _MsG, int _Size, int _bombAmount) {
 		MsG = _MsG;
 		Size = _Size;
@@ -65,7 +64,6 @@ public class Playfield {
 	}
 
 	public void reset() {
-		JOptionPane.showMessageDialog(MsG,"KABOOM! Try again!");
 		cellsFlooded = 0;
 		for (int i = 0; i < Size; i++) {
 			for (int j = 0; j < Size; j++) {
@@ -126,15 +124,15 @@ public class Playfield {
 
 		cells[row][column].update();
 	}
-	
+
 	public void cellFlooded() {
 		cellsFlooded++;
-		if(cellsFlooded >= Size * Size - bombAmount) {
+		if (cellsFlooded >= Size * Size - bombAmount) {
 			JOptionPane.showMessageDialog(MsG, "You won, congratulations!");
-			System.exit(0);
+			reset();
 		}
 	}
-	
+
 	public void cellDried() {
 		cellsFlooded--;
 	}