diff --git a/__init__.py b/__init__.py
index e6edefc..9df9453 100644
--- a/__init__.py
+++ b/__init__.py
@@ -10,6 +10,5 @@ __author__ = """Daniel Martinek"""
 __email__ = "danda.martinek@gmail.com"
 __version__ = "0.0.1"
 
-from .src.more_math.nodes import comfy_entrypoint
 
 
diff --git a/src/more_math/NoiseMathNode.py b/src/more_math/NoiseMathNode.py
index 9e4f4c1..d772dfa 100644
--- a/src/more_math/NoiseMathNode.py
+++ b/src/more_math/NoiseMathNode.py
@@ -1,5 +1,4 @@
 from inspect import cleandoc
-from math import e
 
 from antlr4 import CommonTokenStream, InputStream
 import torch
@@ -10,7 +9,7 @@ from .Parser.MathExprParser import MathExprParser
 from .Parser.MathExprLexer import MathExprLexer
 from .Parser.TensorEvalVisitor import TensorEvalVisitor
 
-from comfy_api.latest import ComfyExtension, io
+from comfy_api.latest import io
 
 # try to import NestedTensor type if available
 import comfy.nested_tensor as _nested_tensor_module
diff --git a/src/more_math/Parser/FloatEvalVisitor.py b/src/more_math/Parser/FloatEvalVisitor.py
index e76a464..46cc932 100644
--- a/src/more_math/Parser/FloatEvalVisitor.py
+++ b/src/more_math/Parser/FloatEvalVisitor.py
@@ -1,7 +1,4 @@
 import math
-from tkinter import SE
-from sympy import true
-import torch
 from .MathExprVisitor import MathExprVisitor
 
 class FloatEvalVisitor(MathExprVisitor):
diff --git a/src/more_math/Parser/MathExprListener.py b/src/more_math/Parser/MathExprListener.py
index 87a3969..c5b9706 100644
--- a/src/more_math/Parser/MathExprListener.py
+++ b/src/more_math/Parser/MathExprListener.py
@@ -1,5 +1,5 @@
 # Generated from MathExpr.g4 by ANTLR 4.13.2
-from antlr4 import *
+from antlr4 import ParseTreeListener
 if "." in __name__:
     from .MathExprParser import MathExprParser
 else:
diff --git a/src/more_math/Parser/MathExprParser.py b/src/more_math/Parser/MathExprParser.py
index 7bd6846..e12dc8d 100644
--- a/src/more_math/Parser/MathExprParser.py
+++ b/src/more_math/Parser/MathExprParser.py
@@ -1,7 +1,6 @@
 # Generated from src/more_math/Parser/MathExpr.g4 by ANTLR 4.13.2
 # encoding: utf-8
-from antlr4 import *
-from io import StringIO
+from antlr4 import ATNDeserializer,Parser,PredictionContextCache,Token,TokenStream,DFA,ParserATNSimulator,ATN, RecognitionException,ParserRuleContext,ParseTreeVisitor,NoViableAltException,RuleContext
 import sys
 if sys.version_info[1] > 5:
 	from typing import TextIO