diff --git a/sea.go b/sea.go
index d128cf7..dcee233 100644
--- a/sea.go
+++ b/sea.go
@@ -1,18 +1,18 @@
-package seaweed
-
 /***
 Пакет seaweed предназначен для облегчения работы с БД sea
 и weed.
-Текущая версия: v1.0.6
+Текущая версия: v1.0.7
 ***/
 
+package seaweed
+
 import (
-	"os"
-	"fmt"
 	"errors"
 	"flag"
+	"fmt"
+	"os"
 
-	_ "github.com/go-sql-driver/mysql" 
+	_ "github.com/go-sql-driver/mysql"
 	"github.com/jmoiron/sqlx"
 
 	"github.com/kataras/pio"
@@ -24,22 +24,23 @@ var p = pio.NewTextPrinter("color", os.Stdout)
 // ============= SEA ============
 
 type Sea struct {
-	db *sqlx.DB
+	db    *sqlx.DB
 	debug bool
 }
 
+// Connect to database and connect to weed
 func Connect() (*Sea, error) {
 	debug := flag.Bool("debug", false, "enable debug mode")
 	fmt.Printf("debug: %v\n", *debug)
 
 	user := os.Getenv("SEA_USER")
 	if user == "" {
-		return nil, errors.New("Не установлена переменная SEA_USER")
+		return nil, errors.New("не установлена переменная SEA_USER")
 	}
 
 	password := os.Getenv("SEA_PASSWORD")
 	if password == "" {
-		return nil, errors.New("Не установлена переменная SEA_PASSWORD")
+		return nil, errors.New("не установлена переменная SEA_PASSWORD")
 	}
 
 	db, err := openDB(user, password)
@@ -50,7 +51,7 @@ func Connect() (*Sea, error) {
 	if *debug {
 		p.Println(pio.Rich("БД sea успешно открыта", pio.Blue))
 	}
-	
+
 	sea := Sea{db: db, debug: *debug}
 	return &sea, nil
 }
@@ -84,4 +85,3 @@ func (s *Sea) Albums() ([]Album, error) {
 	}
 	return albums, nil
 }
-
diff --git a/sea_test.go b/sea_test.go
new file mode 100644
index 0000000..edd5613
--- /dev/null
+++ b/sea_test.go
@@ -0,0 +1,36 @@
+/***
+Пакет seaweed предназначен для облегчения работы с БД sea
+и weed.
+Текущая версия: v1.0.7
+***/
+
+package seaweed
+
+import (
+	"reflect"
+	"testing"
+
+	_ "github.com/go-sql-driver/mysql"
+)
+
+func TestConnect(t *testing.T) {
+	tests := []struct {
+		name    string
+		want    *Sea
+		wantErr bool
+	}{
+		// TODO: Add test cases.
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := Connect()
+			if (err != nil) != tt.wantErr {
+				t.Errorf("Connect() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("Connect() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}