diff --git a/src/main/java/BattleShip/AIGridGUI.java b/src/main/java/BattleShip/AIGridGUI.java index e67f2de..f9a75ef 100644 --- a/src/main/java/BattleShip/AIGridGUI.java +++ b/src/main/java/BattleShip/AIGridGUI.java @@ -252,6 +252,15 @@ public class AIGridGUI extends JPanel { } up.setCell(u); up.setCount(upCount); + + int d = f; + int downCount = -1; + while(d >= 0 && cellsHit[d] && !cellsKilled[d]) { + d = moveDown(d); + downCount++; + } + down.setCell(d); + down.setCount(downCount); } }