update

src/main/java/BattleShip/BSButton.java

@@ -15,4 +15,12 @@ public class BSButton extends JButton {
         return gridLocation;
     }
 
+    public void setCellContents(Ship s) {
+        cellContents = s;
+    }
+
+    public Ship getCellContents() {
+        return cellContents;
+    }
+
 }