diff --git a/src/main/java/BattleShip/BSButton.java b/src/main/java/BattleShip/BSButton.java
index b6c2426..3262b23 100644
--- a/src/main/java/BattleShip/BSButton.java
+++ b/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;
+    }
+
 }
\ No newline at end of file