diff --git a/src/main/java/BattleShip/BSButton.java b/src/main/java/BattleShip/BSButton.java
index 130af7e..7042215 100644
--- a/src/main/java/BattleShip/BSButton.java
+++ b/src/main/java/BattleShip/BSButton.java
@@ -1,5 +1,10 @@
 package BattleShip;
 
-public class BSButton {
+import javax.swing.*;
+
+public class BSButton  {
+
+    int gridLocation;
+    Ship cellContents = null;
 
 }
\ No newline at end of file