diff --git a/src/main/java/BattleShip/BSButton.java b/src/main/java/BattleShip/BSButton.java
index c9eb96e..b6c2426 100644
--- a/src/main/java/BattleShip/BSButton.java
+++ b/src/main/java/BattleShip/BSButton.java
@@ -7,4 +7,12 @@ public class BSButton extends JButton {
     int gridLocation;
     Ship cellContents = null;
 
+    public void setGridLocation(int l) {
+        gridLocation = l;
+    }
+
+    public int getGridLocation() {
+        return gridLocation;
+    }
+
 }
\ No newline at end of file