diff --git a/src/main/java/BattleShip/AIGridGUI.java b/src/main/java/BattleShip/AIGridGUI.java
index 0bfc9f9..fc1e823 100644
--- a/src/main/java/BattleShip/AIGridGUI.java
+++ b/src/main/java/BattleShip/AIGridGUI.java
@@ -138,9 +138,15 @@ public class AIGridGUI extends JPanel {
             testLocations = null;
         }
 
-        //
-
+        //Mark all cells containing a ship and disable all cells in the grid.
 
+        for (BSButton bsb : buttons) {
+            if(bsb.getCellContents() != null) {
+                bsb.setBackground(Color.blue);
+                bsb.setBorder(loweredBevel);
+            }
+            bsb.setEnabled(false);
+        }
 
         text = "Ready to start the game.";
         shipsPlaced = true;