1 files changed, 10 insertions, 14 deletions
diff --git a/src/main/java/Main.java b/src/main/java/Main.java
index 636cd3b..a078bba 100644
--- a/src/main/java/Main.java
+++ b/src/main/java/Main.java
@@ -1,28 +1,24 @@
 import audio.Music;
 import net.dv8tion.jda.api.events.session.ReadyEvent;
 import net.dv8tion.jda.api.hooks.ListenerAdapter;
-import org.json.simple.JSONObject;
-import org.json.simple.parser.JSONParser;
-import java.io.FileReader;
-
+import io.github.cdimascio.dotenv.Dotenv;
 
 public class Main extends ListenerAdapter {
+    private static final Dotenv dotenv = Dotenv.load();
 
-
-    public static void main( String[] args)
-    {
-        new Music("$",readSetting("YOUTUBEAPIKEY"),readSetting("DISCORDTOKEN"));
+    public static void main(String[] args) {
+        new Music("$",
+                readSetting("YOUTUBEAPIKEY"),
+                readSetting("DISCORDTOKEN")
+        );
     }
 
-    public static String readSetting(String parameter){
+    public static String readSetting(String parameter) {
         String value = System.getenv(parameter);
-        if (value != null) {
-            return value;
-        }
-        throw new RuntimeException("Environment variable " + parameter + " not found");
+        if (value != null) return value;
+        return dotenv.get(parameter);
     }
 
-
     @Override
     public void onReady(ReadyEvent event) {
         System.out.println("Loading Complete");