From ebc9c69ded80bf3f5eaa4dced5d1fcfc95b83fad Mon Sep 17 00:00:00 2001
From: Trixkz <loganmagnan@gmail.com>
Date: Tue, 23 Nov 2021 08:36:30 -0500
Subject: [PATCH] t

---
 src/main/java/rip/tilly/bedwars/game/arena/Arena.java     | 6 ++++++
 src/main/java/rip/tilly/bedwars/generators/Generator.java | 1 +
 2 files changed, 7 insertions(+)

diff --git a/src/main/java/rip/tilly/bedwars/game/arena/Arena.java b/src/main/java/rip/tilly/bedwars/game/arena/Arena.java
index 42b73f1..458b936 100644
--- a/src/main/java/rip/tilly/bedwars/game/arena/Arena.java
+++ b/src/main/java/rip/tilly/bedwars/game/arena/Arena.java
@@ -4,6 +4,9 @@ import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
 import lombok.Setter;
+import org.bukkit.Bukkit;
+import rip.tilly.bedwars.generators.GeneratorTier;
+import rip.tilly.bedwars.utils.CC;
 import rip.tilly.bedwars.utils.CustomLocation;
 
 import java.util.ArrayList;
@@ -42,6 +45,9 @@ public class Arena {
     private List<CustomLocation> diamondGenerators = new ArrayList<>();
     private List<CustomLocation> emeraldGenerators = new ArrayList<>();
 
+    private GeneratorTier diamondGeneratorTier = GeneratorTier.ONE;
+    private GeneratorTier emeraldGeneratorTier = GeneratorTier.ONE;
+
     private boolean enabled;
 
     public CopiedArena getAvailableArena() {
diff --git a/src/main/java/rip/tilly/bedwars/generators/Generator.java b/src/main/java/rip/tilly/bedwars/generators/Generator.java
index f501783..4eee484 100644
--- a/src/main/java/rip/tilly/bedwars/generators/Generator.java
+++ b/src/main/java/rip/tilly/bedwars/generators/Generator.java
@@ -1,5 +1,6 @@
 package rip.tilly.bedwars.generators;
 
+import org.bukkit.Bukkit;
 import org.bukkit.Location;
 import org.bukkit.Material;
 import org.bukkit.entity.ArmorStand;