diff --git a/cmd/new.go b/cmd/new.go
index b24bef6..fa8c95f 100644
--- a/cmd/new.go
+++ b/cmd/new.go
@@ -1,9 +1,10 @@
 package cmd
 
 import (
+	"log"
+
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
-	"log"
 
 	"password-generator/passGen"
 )
diff --git a/main_test.go b/passGen/passGen_test.go
similarity index 97%
rename from main_test.go
rename to passGen/passGen_test.go
index 3de86fb..b99faf9 100644
--- a/main_test.go
+++ b/passGen/passGen_test.go
@@ -1,10 +1,11 @@
-package main
+package passGen_test
 
 import (
 	"math/rand"
-	"password-generator/passGen"
 	"strings"
 	"testing"
+
+	"password-generator/passGen"
 )
 
 func TestGeneratePasswordWithGivenCharacterNumber(t *testing.T) {