diff --git a/scaffoldforge/cli/__init__.py b/scaffoldforge/cli/__init__.py
index 18a3f79..1238c71 100644
--- a/scaffoldforge/cli/__init__.py
+++ b/scaffoldforge/cli/__init__.py
@@ -1,16 +1,22 @@
 """CLI module for ScaffoldForge."""
 
-import sys
-from pathlib import Path
-
 import click
 
-from scaffoldforge.cli.commands import generate, preview, list_templates
+from scaffoldforge.cli.commands import generate, list_templates, preview
 from scaffoldforge.config import get_config
-from scaffoldforge.parsers import IssueParser
-from scaffoldforge.generators import StructureGenerator, CodeGenerator
+from scaffoldforge.generators import CodeGenerator, StructureGenerator
 from scaffoldforge.templates import TemplateEngine
 
+__all__ = [
+    "cli",
+    "generate",
+    "preview",
+    "list_templates",
+    "StructureGenerator",
+    "CodeGenerator",
+    "TemplateEngine",
+]
+
 
 @click.group()
 @click.option("--verbose", "-v", is_flag=True, help="Enable verbose output")