diff --git a/perlinator.py b/perlinator.py
index 0309ade..3c6a869 100755
--- a/perlinator.py
+++ b/perlinator.py
@@ -5,65 +5,43 @@ from tkinter import ttk
 from PIL import Image, ImageTk
 import os
 from os import listdir
+from tkinter import filedialog
 
 parent = tk.Tk()
 parent.title("Perlinator")
 parent.minsize(320, 240)
 
 
-def button_clicked():
-    global isittrue
-    if isittrue == 0:
-        image2 = Image.open("test/tux.jpg")
-        image2 = ImageTk.PhotoImage(image2)
-        image_label.configure(image=image2)
-        image_label.image = image2
-        isittrue = 1
+def upload_folder():
+    folder_path = filedialog.askdirectory()
+    if folder_path:
+        image_files = [
+            f
+            for f in listdir(folder_path)
+            if f.lower().endswith((".png", ".jpg", ".jpeg", ".gif"))
+        ]
+        print(f"Found {len(image_files)} images in {folder_path}:")
+        for image_file in image_files:
+            print(image_file)
+
+        open_button.pack(pady=10)
+
     else:
-        image_label.configure(image=img)
-        image_label.image = img
-        isittrue = 0
+        popup = tk.Toplevel(parent)
+        popup.title("No Folder Selected")
+        popup.geometry("300x100")
+        label = tk.Label(popup, text="No folder selected.")
+        label.pack(pady=20)
+        popup.after(2000, popup.destroy)
 
 
-isittrue = 0
-
-button = tk.Button(
+open_button = tk.Button(
     parent,
-    text="Click Me",
-    command=button_clicked,
-    activebackground="blue",
-    activeforeground="white",
     anchor="center",
-    bd=3,
-    bg="lightgray",
     cursor="hand2",
-    disabledforeground="gray",
-    fg="black",
-    font=("Arial", 12),
-    height=2,
-    highlightbackground="black",
-    highlightcolor="green",
-    highlightthickness=2,
-    justify="center",
-    overrelief="raised",
-    padx=10,
-    pady=5,
-    width=15,
-    wraplength=100,
+    text="Selectionner le dossier d'images",
+    command=upload_folder,
 )
-
-button.pack(padx=20, pady=20)
-
-folder_dir = "./test/"
-for images in os.listdir(folder_dir):
-
-    if images.endswith(".png") or images.endswith(".jpg") or images.endswith(".jpeg"):
-        print(images)
-
-img = Image.open("test/logo.png")
-img = ImageTk.PhotoImage(img)
-
-image_label = tk.Label(parent, image=img)
-image_label.pack()
+open_button.pack(pady=10)
 
 parent.mainloop()