diff --git a/src/main/java/BattleShip/GridGUI.java b/src/main/java/BattleShip/GridGUI.java
index 9b8c21a..5a9b32a 100644
--- a/src/main/java/BattleShip/GridGUI.java
+++ b/src/main/java/BattleShip/GridGUI.java
@@ -80,6 +80,11 @@ public class GridGUI {
     public boolean getEndGame() {
         return endGame;
     }
+    public void setEndGame() {
+        for(JButton j : buttons){
+            j.setEnabled(false);
+        }
+    }
     public class MyCellListener implements ActionListener {
         public void actionPerformed(ActionEvent a) {
             if(!clicked) {