diff --git a/.github/workflows/go.yml b/.github/workflows/go.yml
index 56190bb..0482c10 100644
--- a/.github/workflows/go.yml
+++ b/.github/workflows/go.yml
@@ -24,3 +24,17 @@ jobs:
     - name: Build
       run: go build -v ./...
 
+  docker:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2
+
+      - name: Build Docker image
+        run: |
+          docker build -t todo-list:latest .
+
+      - name: Run Docker container
+        run: |
+          docker run --rm to-do-list:latest
diff --git a/go.mod b/go.mod
index 0472b77..5440abf 100644
--- a/go.mod
+++ b/go.mod
@@ -3,7 +3,6 @@ module github.com/KingLeak95/todo-list-go
 go 1.21.0
 
 require (
-	github.com/DATA-DOG/go-sqlmock v1.5.0
 	github.com/gin-gonic/gin v1.9.1
 	gorm.io/driver/sqlite v1.5.3
 	gorm.io/gorm v1.25.4
diff --git a/go.sum b/go.sum
index e8a558b..6b71f0a 100644
--- a/go.sum
+++ b/go.sum
@@ -1,5 +1,3 @@
-github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60=
-github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM=
 github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
 github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s=
 github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
diff --git a/main.go b/main.go
index 99b8e9c..7e38e06 100644
--- a/main.go
+++ b/main.go
@@ -1,15 +1,11 @@
 package main
 
 import (
+	"github.com/KingLeak95/todo-list-go/models"
 	"github.com/gin-gonic/gin"
 	"net/http"
-	"github.com/KingLeak95/todo-list-go/models"
 )
 
-func createUser(c *gin.Context) {
-
-}
-
 func main() {
 	r := gin.Default()
 
diff --git a/main_test.go b/main_test.go
deleted file mode 100644
index 1ecd3d4..0000000
--- a/main_test.go
+++ /dev/null
@@ -1,34 +0,0 @@
-package main
-import (
-  "github.com/gin-gonic/gin"
-  "github.com/DATA-DOG/go-sqlmock"
-  "testing"
-)
-
-func SetUpRouter() *gin.Engine{
-    router := gin.Default()
-    return router
-}
-
-func TestCreateUser(t *testing.T) {
-    db, mock, err := sqlmock.New()
-	if err != nil {
-		t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
-	}
-	defer db.Close()
-    r := SetUpRouter()
-    r.POST("/company", NewCompanyHandler)
-    companyId := xid.New().String()
-    company := Company{
-        ID: companyId,
-        Name: "Demo Company",
-        CEO: "Demo CEO",
-        Revenue: "35 million",
-    }
-    jsonValue, _ := json.Marshal(company)
-    req, _ := http.NewRequest("POST", "/company", bytes.NewBuffer(jsonValue))
-
-    w := httptest.NewRecorder()
-    r.ServeHTTP(w, req)
-    assert.Equal(t, http.StatusCreated, w.Code)
-}