diff --git a/CROP_OBJECT/__pycache__/__init__.cpython-310.pyc b/CROP_OBJECT/__pycache__/__init__.cpython-310.pyc
index 1d299ba..1a6576e 100644
Binary files a/CROP_OBJECT/__pycache__/__init__.cpython-310.pyc and b/CROP_OBJECT/__pycache__/__init__.cpython-310.pyc differ
diff --git a/CROP_OBJECT/__pycache__/crop.cpython-310.pyc b/CROP_OBJECT/__pycache__/crop.cpython-310.pyc
index c714720..79e59d5 100644
Binary files a/CROP_OBJECT/__pycache__/crop.cpython-310.pyc and b/CROP_OBJECT/__pycache__/crop.cpython-310.pyc differ
diff --git a/CROP_OBJECT/crop.py b/CROP_OBJECT/crop.py
index 70c646a..cce9ee9 100644
--- a/CROP_OBJECT/crop.py
+++ b/CROP_OBJECT/crop.py
@@ -37,8 +37,6 @@ class ObjectCrop:
 
         return (cropped, )
 
-# A dictionary that contains all nodes you want to export with their names
-# NOTE: names should be globally unique
 NODE_CLASS_MAPPINGS = {
     "ObjectCrop": ObjectCrop
 }
diff --git a/Custom_comfyui_llm_assistant/AssistantNodes.py b/Custom_comfyui_llm_assistant/AssistantNodes.py
index e0b558b..70882bb 100644
--- a/Custom_comfyui_llm_assistant/AssistantNodes.py
+++ b/Custom_comfyui_llm_assistant/AssistantNodes.py
@@ -64,12 +64,12 @@ class GenerateStableDiffsutionPromptLLM:
                 "stop": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "response_pattern": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "temperature": ("FLOAT", {
                     "default": 1.0,
@@ -98,7 +98,7 @@ class GenerateStableDiffsutionPromptLLM:
 
     FUNCTION = "generateStableDiffsutionPrompt"
 
-    CATEGORY = "cropcropcropcrop"
+    CATEGORY = "LLM"
 
     def generateStableDiffsutionPrompt(self, object1, desc_obj1, object2, desc_obj2 ,template_system, template_user, stop, response_pattern,  temperature, max_tokens, model_name):
         config = load_config()
@@ -170,12 +170,12 @@ class TranslateTextLLM:
                 "stop": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "response_pattern": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "temperature": ("FLOAT", {
                     "default": 1.0,
@@ -274,12 +274,12 @@ class ChatWithLLM:
                 "stop": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "response_pattern": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "temperature": ("FLOAT", {
                     "default": 1.0,
@@ -329,12 +329,12 @@ class ChatWithLLM:
 
 NODE_CLASS_MAPPINGS = {
     "Generate Stable Diffsution Prompt With LLM duci": GenerateStableDiffsutionPromptLLM,
-    # "Translate Text With LLM": TranslateTextLLM,
-    # "Chat With LLM": ChatWithLLM,
+    "Translate Text With LLM": TranslateTextLLM,
+    "Chat With LLM": ChatWithLLM,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "Generate Stable Diffsution Prompt With LLM duci": "HUHU",
-    # "Translate Text With LLM": "Translate Text With LLM",
-    # "Chat With LLM": "Chat With LLM",
+    "Generate Stable Diffsution Prompt With LLM duci": "Custom Generate Stable Diffsution Prompt With LLM",
+    "Translate Text With LLM": "Translate Text With LLM",
+    "Chat With LLM": "Chat With LLM",
 }
diff --git a/README.md b/README.md
index c87a0a6..e9bc31b 100644
--- a/README.md
+++ b/README.md
@@ -51,7 +51,6 @@ _All the folders, except `CROP_OBJECT`, are from other repositories. However, I
 - **[comfyui-llm-assistant](https://github.com/longgui0318/comfyui-llm-assistant)**
   - Removed input field: `prompt`.
   - Added 4 input fields: `object1`, `desc_obj1`, `object2`, `desc_obj2`.
-  - Change all `default` value to `""` because the newest frontend of ComfyUI do not accept `None`
 
 ## 🤝 Contributing
 Contributions are welcome! Please open an issue or submit a pull request.
diff --git a/__init__.py b/__init__.py
index bee3c70..89304a0 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,14 +1,7 @@
-# from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_EfficientSAM
-# from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_SEGS
+from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_EfficientSAM
+from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_SEGS
 from .Custom_comfyui_llm_assistant import AssistantNodes
 from .CROP_OBJECT import crop
 
-# NODE_CLASS_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_CLASS_MAPPINGS, **YOLO_WORLD_SEGS.NODE_CLASS_MAPPINGS, **AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
-# NODE_DISPLAY_NAME_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_DISPLAY_NAME_MAPPINGS, **YOLO_WORLD_SEGS.NODE_DISPLAY_NAME_MAPPINGS, **AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}
-
-
-NODE_CLASS_MAPPINGS = {**AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
-NODE_DISPLAY_NAME_MAPPINGS = {**AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}
-
-print(NODE_CLASS_MAPPINGS)
-print(NODE_DISPLAY_NAME_MAPPINGS)
\ No newline at end of file
+NODE_CLASS_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_CLASS_MAPPINGS, **YOLO_WORLD_SEGS.NODE_CLASS_MAPPINGS, **AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
+NODE_DISPLAY_NAME_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_DISPLAY_NAME_MAPPINGS, **YOLO_WORLD_SEGS.NODE_DISPLAY_NAME_MAPPINGS, **AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}
\ No newline at end of file