1 files changed, 10 insertions, 4 deletions

diff --git a/src/main.rs b/src/main.rs
index d177c3e..32727c7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,15 +1,20 @@
 use std::{env, process};
 
-use modlist::{config::Cfg, input::{get_input, Cmd}, update, download, list, io, modification};
+use modlist::{config::Cfg, input::{get_input, Cmd}, update, download, list, io, modification, setup};
 
 #[tokio::main]
 async fn main() {
     let config = Cfg::init("modlist.toml").unwrap();
-
+    
     let mut args: Vec<String> = env::args().collect();
     args.reverse();
     args.pop();
     args.reverse();
+
+    if args.is_empty() {
+        println!("Please enter an argument");
+        process::exit(1);
+    };
     
     let input = match get_input(config.clone(), args).await {
         Ok(i) => i,
@@ -19,8 +24,6 @@ async fn main() {
         }
     };
 
-    //dbg!(&input);
-
     match input.clone().command.unwrap() {
         Cmd::Mod => {
             modification(config, input).await
@@ -41,6 +44,9 @@ async fn main() {
             show_version();
             Ok(())
         },
+        Cmd::Setup => {
+            setup(config).await
+        },
     }.unwrap()
 }