1 files changed, 15 insertions, 6 deletions
diff --git a/src/main/java/me/xmrvizzy/skyblocker/gui/ContainerSolver.java b/src/main/java/me/xmrvizzy/skyblocker/gui/ContainerSolver.java
index 84de64eb..e52855a2 100644
--- a/src/main/java/me/xmrvizzy/skyblocker/gui/ContainerSolver.java
+++ b/src/main/java/me/xmrvizzy/skyblocker/gui/ContainerSolver.java
@@ -1,5 +1,6 @@
 package me.xmrvizzy.skyblocker.gui;
 
+import net.minecraft.client.gui.screen.ingame.GenericContainerScreen;
 import net.minecraft.item.ItemStack;
 
 import java.util.List;
@@ -11,22 +12,30 @@ import java.util.regex.Pattern;
  */
 public abstract class ContainerSolver {
     private final Pattern CONTAINER_NAME;
-    protected final static int GREEN_HIGHLIGHT = 128 << 24 | 64 << 16 | 196 << 8 | 64;
-    protected final static int GRAY_HIGHLIGHT = 128 << 24 | 64 << 16 | 64 << 8 | 64;
+    protected static final int RED_HIGHLIGHT = 64 << 24 | 255 << 16;
+    protected static final int YELLOW_HIGHLIGHT = 128 << 24 | 255 << 16 | 255 << 8;
+    protected static final int GREEN_HIGHLIGHT = 128 << 24 | 64 << 16 | 196 << 8 | 64;
+    protected static final int GRAY_HIGHLIGHT = 128 << 24 | 64 << 16 | 64 << 8 | 64;
 
-    public ContainerSolver(String containerName) {
+    protected ContainerSolver(String containerName) {
         CONTAINER_NAME = Pattern.compile(containerName);
     }
 
-    public abstract boolean isEnabled();
+    protected abstract boolean isEnabled();
 
     public Pattern getName() {
         return CONTAINER_NAME;
     }
 
-    public abstract List<ColorHighlight> getColors(String[] groups, Map<Integer, ItemStack> slots);
+    protected void start(GenericContainerScreen screen) {
+    }
+
+    protected void reset() {
+    }
+
+    protected abstract List<ColorHighlight> getColors(String[] groups, Map<Integer, ItemStack> slots);
 
-    public void trimEdges(Map<Integer, ItemStack> slots, int rows) {
+    protected void trimEdges(Map<Integer, ItemStack> slots, int rows) {
         for (int i = 0; i < rows; i++) {
             slots.remove(9 * i);
             slots.remove(9 * i + 8);