diff --git a/apt-install.sh b/apt-install.sh
index 847fe00..034af9f 100755
--- a/apt-install.sh
+++ b/apt-install.sh
@@ -1,17 +1,24 @@
 #!/bin/bash
-
+set -e
 
 get_all_dependencies() {
     local apt_package=$1
-    apt-rdepends $apt_package | grep -v "^ "
+    apt depends $apt_package | grep -Eo "\s[a-z].+"
 }
 
 download_packages() {
-    local deps=$* 
+    local deps=$*
+    echo $deps
     for package_name in ${deps[@]};
     do
+        echo $package_name
         apt download $package_name
     done
 }
 
-download_packages $(get_all_dependencies apt-offline)
\ No newline at end of file
+main() {
+    # echo $(get_all_dependencies $1)
+    download_packages $(get_all_dependencies $1)
+}
+
+main $@
\ No newline at end of file