diff --git a/src/main/java/BattleShip/AIGridGUI.java b/src/main/java/BattleShip/AIGridGUI.java
index 1cf8be4..f3d9bfe 100644
--- a/src/main/java/BattleShip/AIGridGUI.java
+++ b/src/main/java/BattleShip/AIGridGUI.java
@@ -1,11 +1,9 @@
 package BattleShip;
 
 import javax.swing.*;
-import javax.swing.border.Border;
+import javax.swing.border.*;
 import java.awt.*;
-import java.awt.event.MouseEvent;
-import java.awt.event.MouseListener;
-import java.util.ArrayList;
+import java.awt.event.*;
 import java.util.*;
 
 
diff --git a/src/main/java/BattleShip/BShip.java b/src/main/java/BattleShip/BShip.java
index 032d572..89cd601 100644
--- a/src/main/java/BattleShip/BShip.java
+++ b/src/main/java/BattleShip/BShip.java
@@ -150,6 +150,7 @@ public class BShip {
                             }
                         } else {
                             textArea.setText(grid2.getText());
+                            }
                             if(grid1.getClicked()) {
 
                                 textArea.setText(grid1.getText());
@@ -169,7 +170,7 @@ public class BShip {
 
                                 grid1.setClicked();
                             }
-                        }
+
                 }
             }
         }
@@ -182,8 +183,7 @@ public class BShip {
                     synchronized (lock) {
                        try {
                             lock.wait();
-                        } catch (InterruptedException e) {
-                        }
+                        } catch (InterruptedException e) {}
                     }
                 }
                 grid1.enableCells();
diff --git a/target/classes/BattleShip/AIGridGUI$DirectionCompare.class b/target/classes/BattleShip/AIGridGUI$DirectionCompare.class
index a226425..411fa71 100644
Binary files a/target/classes/BattleShip/AIGridGUI$DirectionCompare.class and b/target/classes/BattleShip/AIGridGUI$DirectionCompare.class differ
diff --git a/target/classes/BattleShip/AIGridGUI$SetShipsListener.class b/target/classes/BattleShip/AIGridGUI$SetShipsListener.class
index a43d2b1..9334499 100644
Binary files a/target/classes/BattleShip/AIGridGUI$SetShipsListener.class and b/target/classes/BattleShip/AIGridGUI$SetShipsListener.class differ
diff --git a/target/classes/BattleShip/AIGridGUI.class b/target/classes/BattleShip/AIGridGUI.class
index 7c2fc5b..f16ca3f 100644
Binary files a/target/classes/BattleShip/AIGridGUI.class and b/target/classes/BattleShip/AIGridGUI.class differ
diff --git a/target/classes/BattleShip/BShip$IncomingGuessReader.class b/target/classes/BattleShip/BShip$IncomingGuessReader.class
index a8f7d4c..79ec450 100644
Binary files a/target/classes/BattleShip/BShip$IncomingGuessReader.class and b/target/classes/BattleShip/BShip$IncomingGuessReader.class differ
diff --git a/target/classes/BattleShip/BShip$ShipsPlacedReader.class b/target/classes/BattleShip/BShip$ShipsPlacedReader.class
index 1893eb1..85b7f4d 100644
Binary files a/target/classes/BattleShip/BShip$ShipsPlacedReader.class and b/target/classes/BattleShip/BShip$ShipsPlacedReader.class differ