diff --git a/src/main/java/BattleShip/Direction.java b/src/main/java/BattleShip/Direction.java
index 785838f..29caf4c 100644
--- a/src/main/java/BattleShip/Direction.java
+++ b/src/main/java/BattleShip/Direction.java
@@ -13,4 +13,11 @@ public class Direction {
         return countHits;
     }
 
+    public void setCell(int b) {
+        cellLocation = b;
+    }
+
+    public int getCell() {
+        return cellLocation;
+    }
 }