diff --git a/setup.sh b/setup.sh
index eb01f21..d79f9f3 100644
--- a/setup.sh
+++ b/setup.sh
@@ -63,9 +63,11 @@ function config() {
     if [ $? -eq 0 ]; then
         echo "Selected file: $WALLPAPER_FILE"
         if [ -f "$WALLPAPER_FILE" ]; then
-            echo "Creating configuration file"
+            echo "+ Creating configuration directory"
+            mkdir -p ~/.config/Wallify
+            echo "+ Creating configuration file"
             echo "WALLPAPER_FILE=$WALLPAPER_FILE" > ~/.config/Wallify/wallpaper.conf
-            echo "Creating start.sh script"
+            echo "+ Creating start.sh script"
             cat > ~/.config/Wallify/start.sh <<EOF
 #!/bin/bash
 user=$(whoami)
@@ -76,7 +78,7 @@ gsettings set org.gnome.desktop.background picture-uri file://$WALLPAPER_FILE
 /home/UCA/machirat1/public/WallPaper/wallify -ni -fs -s -sp -st -b -nf -- mpv -wid WID --loop --no-audio $WALLPAPER_FILE &
 EOF
             chmod +x start.sh
-            echo "Creating autostart file"
+            echo "+ Creating autostart file"
             cat >  ~/.config/autostart/wallify.sh.desktop <<EOF
 [Desktop Entry]
 Version=1.0
@@ -85,7 +87,7 @@ Name=Wallify
 Exec=/home/UCA/$USER/.config/Wallify/start.sh
 Comment=Wallify
 EOF
-            echo "Setup complete. You can now run ./start.sh"
+            echo "Setup complete. You can restart your session to apply the changes."
         else
             zenity --error --text="Selected file is not a valid file"
         fi