diff --git a/LCM/__init__.py b/LCM/__init__.py
index e25e605..e69de29 100644
--- a/LCM/__init__.py
+++ b/LCM/__init__.py
@@ -1,3 +0,0 @@
-from .LCM_Nodes import NODE_CLASS_MAPPINGS
-
-__all__ = ['NODE_CLASS_MAPPINGS']
diff --git a/LCM/LCM_Nodes.py b/LCM_Nodes.py
similarity index 95%
rename from LCM/LCM_Nodes.py
rename to LCM_Nodes.py
index 5c407f3..8ab2570 100644
--- a/LCM/LCM_Nodes.py
+++ b/LCM_Nodes.py
@@ -1,56 +1,20 @@
-from tkinter import Y
-import torch
-from PIL import Image, ImageDraw, ImageFont, ImageEnhance, ImageOps, ImageFilter
-import PIL
-import torch.nn.functional as F
-import cv2
-import os
-import torch
-import sys
+from PIL import ImageDraw, ImageOps, ImageFilter
 import json
-import hashlib
-import traceback
-import time
-import re
-import glob
+
 from PIL.PngImagePlugin import PngInfo
 import numpy as np
-import safetensors.torch
-import random
-sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy"))
-
-
-import comfy.diffusers_load
-import comfy.samplers
-import comfy.sample
-import comfy.sd
-import comfy.utils
-
-import comfy.clip_vision
-
-import comfy.model_management
-import importlib
-
 import folder_paths
-import latent_preview
-
-import math
-import struct
-
-from .lcm_pipeline_inpaint import LatentConsistencyModelPipeline_inpaint, LCMScheduler_X
+from .LCM.lcm_pipeline_inpaint import LatentConsistencyModelPipeline_inpaint, LCMScheduler_X
 from diffusers import AutoencoderKL, UNet2DConditionModel
 #from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
 from transformers import CLIPTokenizer, CLIPTextModel, CLIPImageProcessor
 import os
 import torch
-from tqdm import tqdm
-from safetensors.torch import load_file
-from compel import Compel
-import argparse
-import random
 from PIL import Image
 import tomesd
 
+from comfy.cli_args import args
+
 
 class LCMLoader:
     def __init__(self):
diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..e25e605
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,3 @@
+from .LCM_Nodes import NODE_CLASS_MAPPINGS
+
+__all__ = ['NODE_CLASS_MAPPINGS']