diff --git a/shemum b/shemum
index 141698b..9c95ae1 100755
--- a/shemum
+++ b/shemum
@@ -42,6 +42,15 @@ case $OS_NAME in
   ;;
 esac
 
+# Functions
+
+# Virtual machine removal
+remove function { 
+  rm -r $ROOT_PATH/$OS_NAME 
+}
+
+
+
 # Options
 case $OPTION in
   # Initializes the VM and starts the installer image.
@@ -76,7 +85,7 @@ case $OPTION in
                        -m $RAM -smp $CPU
   ;;
   "remove")
-    rm -r $ROOT_PATH/$OS_NAME
+    remove
   ;;
   *)
     echo "Usage: shemum [init] <virtual machine name> <disk size>"