diff --git a/.DS_Store b/.DS_Store
index 810ef7b..551f66e 100644
Binary files a/.DS_Store and b/.DS_Store differ
diff --git a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/ClassCreator.kt b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/ClassCreator.kt
index 8d6e0e5..0a14210 100644
--- a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/ClassCreator.kt
+++ b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/ClassCreator.kt
@@ -86,7 +86,4 @@ object ClassCreator {
         return newSubdirectory
     }
 
-
-
-
 }
diff --git a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CleanArchitectureTemplateName.kt b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CleanArchitectureTemplateName.kt
index a0cd943..abbaac0 100644
--- a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CleanArchitectureTemplateName.kt
+++ b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CleanArchitectureTemplateName.kt
@@ -8,13 +8,10 @@ import com.intellij.psi.PsiDirectory
 import com.intellij.psi.PsiManager
 import com.intellij.util.ui.JBUI
 import java.awt.Dimension
-import java.awt.FlowLayout
 import java.awt.GridBagConstraints
 import java.awt.GridBagLayout
-import java.awt.Insets
 import javax.swing.Box
 import javax.swing.ButtonGroup
-import javax.swing.JButton
 import javax.swing.JComponent
 import javax.swing.JLabel
 import javax.swing.JPanel
@@ -30,8 +27,8 @@ class CleanArchitectureTemplateName(
     private lateinit var panel: JPanel
 
     private val featureNameTextField: JTextField = JTextField()
-    private val responseNameTextField: JTextField = JTextField()
-    private val bodyNameTextField: JTextField = JTextField()
+//    private val responseNameTextField: JTextField = JTextField()
+//    private val bodyNameTextField: JTextField = JTextField()
 
     private val javaRadioButton: JRadioButton = JRadioButton("Java")
     private val flutterRadioButton: JRadioButton = JRadioButton("Flutter")
diff --git a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CodeGeneratorClass.kt b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CodeGeneratorClass.kt
index 3592f64..74c12f4 100644
--- a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CodeGeneratorClass.kt
+++ b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CodeGeneratorClass.kt
@@ -14,8 +14,7 @@ object CodeGeneratorClass {
         val featureNameLowerCase = featureName!!.replaceFirstChar { it.lowercase() }
         val featureNameUpperCase = featureName.replaceFirstChar { it.uppercase() }
 
-
-        return when (selectedLanguage) {
+         when (selectedLanguage) {
             LanguageSelection.JAVA -> {
                 return """package ${packageName}.${featureNameLowerCase}.ui
                     
diff --git a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CreateCleanArchitectureTemplate.kt b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CreateCleanArchitectureTemplate.kt
index c7e5a1e..ab3d041 100644
--- a/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CreateCleanArchitectureTemplate.kt
+++ b/src/main/kotlin/com/github/dekola/androidcleanarchitecturetemplateplugin/CreateCleanArchitectureTemplate.kt
@@ -16,7 +16,7 @@ class CreateCleanArchitectureTemplate : AnAction() {
             println("Selected directory: ${selectedFile.toNioPath()}")
             println("Selected directory: ${selectedFile.presentableUrl}")
             println("Selected directory: $selectedFile")
-            CleanArchitectureTemplateName(event, selectedFile).show()
+                .CleanArchitectureTemplateName(event, selectedFile).show()
             // Implement the logic you want to perform on the selected directory
         } else {
             // Handle cases where the selection is not a directory