diff --git a/src/main/java/BattleShip/BSButton.java b/src/main/java/BattleShip/BSButton.java
index 7042215..c9eb96e 100644
--- a/src/main/java/BattleShip/BSButton.java
+++ b/src/main/java/BattleShip/BSButton.java
@@ -2,7 +2,7 @@ package BattleShip;
 
 import javax.swing.*;
 
-public class BSButton  {
+public class BSButton extends JButton {
 
     int gridLocation;
     Ship cellContents = null;