1 files changed, 7 insertions, 6 deletions

diff --git a/src/main/java/los/numeros/HidersAndDiggers/modules/MainModule.java b/src/main/java/los/numeros/HidersAndDiggers/modules/MainModule.java
index f941fdf..7513cca 100644
--- a/src/main/java/los/numeros/HidersAndDiggers/modules/MainModule.java
+++ b/src/main/java/los/numeros/HidersAndDiggers/modules/MainModule.java
@@ -2,25 +2,25 @@ package los.numeros.HidersAndDiggers.modules;
 
 import los.numeros.GameEngine.Module;
 import los.numeros.utils.GiveItem;
-import net.minecraft.enchantment.Enchantment;
 import net.minecraft.enchantment.Enchantments;
-import net.minecraft.entity.attribute.EntityAttribute;
-import net.minecraft.entity.attribute.EntityAttributeModifier;
-import net.minecraft.entity.attribute.EntityAttributes;
 import net.minecraft.entity.effect.StatusEffectInstance;
 import net.minecraft.entity.effect.StatusEffects;
 import net.minecraft.item.ItemStack;
 import net.minecraft.nbt.NbtCompound;
+import net.minecraft.server.command.ServerCommandSource;
 import net.minecraft.server.network.ServerPlayerEntity;
+import net.minecraft.server.world.ServerWorld;
 import net.minecraft.world.GameMode;
 
 public class MainModule implements Module {
   ServerPlayerEntity[] diggers, hidders, players;
+  ServerCommandSource source;
 
-  public MainModule(ServerPlayerEntity[] diggers, ServerPlayerEntity[] hidders, ServerPlayerEntity[] players) {
+  public MainModule(ServerPlayerEntity[] diggers, ServerPlayerEntity[] hidders, ServerPlayerEntity[] players, ServerCommandSource source) {
     this.diggers = diggers;
     this.hidders = hidders;
     this.players = players;
+    this.source = source;
   }
 
   public void giveDiggerItems(ServerPlayerEntity[] diggers) {
@@ -60,7 +60,6 @@ public class MainModule implements Module {
     for (ServerPlayerEntity p : hiders) {
       // equip picaxe on each player
       p.equip(0, pickaxe);
-
       // give stone x10
       GiveItem.giveMultipleItem(p, "minecraft:stone", null, 10);
     }
@@ -85,6 +84,8 @@ public class MainModule implements Module {
     }
     giveDiggerItems(diggers);
     giveHidersItems(hidders);
+
+    ServerWorld world = source.getWorld();
     // TODO: FILL AND TP PLAYERS AND CLEAR ALL ITEMS
     return 0;
   }