diff --git a/cleanup.py b/cleanup.py
new file mode 100644
index 0000000..86263b9
--- /dev/null
+++ b/cleanup.py
@@ -0,0 +1,10 @@
+import subprocess
+
+def remove_output_parts_directory():
+    try:
+        subprocess.run(["rm", "-rf", "./output_samples"])
+        print("Successfully removed the output_samples directory.")
+    except Exception as e:
+        print(f"Error removing output_samples directory: {str(e)}")
+
+remove_output_parts_directory()