diff --git a/public/data/python.json b/public/data/python.json
index b4c9c2b0..9a3e92bc 100644
--- a/public/data/python.json
+++ b/public/data/python.json
@@ -131,17 +131,13 @@
           "import os",
           "",
           "def find_files(directory, file_type):",
-          "  file_type = file_type.lower()  # Convert file_type to lowercase",
-          "  found_files = []",
+          "    file_type = file_type.lower()",
+          "    found_files = []",
           "",
-          "  for root, _, files in os.walk(directory):",
-          "    for file in files:",
-          "      file_ext = os.path.splitext(file)[1].lower()",
-          "      if file_ext == file_type:",
-          "        full_path = os.path.join(root, file)",
-          "        found_files.append(full_path)",
-          "",
-          "  return found_files",
+          "    for entry in os.scandir(directory):",
+          "        if entry.is_file() and entry.name.lower().endswith(file_type):",
+          "            found_files.append(entry.name)",
+          "    return found_files",
           "",
           "# Example Usage:",
           "pdf_files = find_files('/path/to/your/directory', '.pdf')",