diff --git a/Train_Supervised_Variational.py b/Train_Supervised_Variational.py
index 71de6f6..16cc707 100644
--- a/Train_Supervised_Variational.py
+++ b/Train_Supervised_Variational.py
@@ -26,10 +26,6 @@
 sys.path.append('src')
 import mineralML as mm
 
-import concurrent.futures
-from multiprocessing import freeze_support
-import itertools 
-
 from matplotlib import pyplot as plt
 from matplotlib import rc
 import seaborn as sns
diff --git a/environment.yml b/environment.yml
index 78907ea..538089c 100644
--- a/environment.yml
+++ b/environment.yml
@@ -24,4 +24,5 @@ dependencies:
     - pip:
         - pyrolite
         - PyIRoGlass
-        - hdbscan
\ No newline at end of file
+        - hdbscan
+        - imblearn
\ No newline at end of file
diff --git a/none.pdf b/none.pdf
deleted file mode 100644
index bd0ec24..0000000
Binary files a/none.pdf and /dev/null differ
diff --git a/setup.py b/setup.py
index a4c842b..13a2f96 100644
--- a/setup.py
+++ b/setup.py
@@ -34,7 +34,8 @@
             'scikit-learn',
             'scipy',
             'torch',
-            'hdbscan'
+            'hdbscan', 
+            'imblearn'
             ],
 
     classifiers=[