diff --git a/go.mod b/go.mod
index a680d05..8efa94d 100644
--- a/go.mod
+++ b/go.mod
@@ -1,4 +1,4 @@
-module github.com/willdot/message-broker
+module github.com/willdot/messagebroker
 
 go 1.21.0
 
diff --git a/server.go b/server.go
index f5337b9..ec85548 100644
--- a/server.go
+++ b/server.go
@@ -1,4 +1,4 @@
-package main
+package messagebroker
 
 import (
 	"context"
@@ -20,7 +20,7 @@ const (
 	Publish     Action = 3
 )
 
-type message struct {
+type Message struct {
 	Topic string `json:"topic"`
 	Data  []byte `json:"data"`
 }
@@ -223,7 +223,7 @@ func (s *Server) handlePublisherConn(conn net.Conn) {
 		return
 	}
 
-	var msg message
+	var msg Message
 	err = json.Unmarshal(buf, &msg)
 	if err != nil {
 		_, _ = conn.Write([]byte("invalid message"))
diff --git a/server_test.go b/server_test.go
index 087774c..33d86a3 100644
--- a/server_test.go
+++ b/server_test.go
@@ -1,4 +1,4 @@
-package main
+package messagebroker
 
 import (
 	"context"
@@ -204,7 +204,7 @@ func TestSendsDataToTopicSubscribers(t *testing.T) {
 
 	// send some data
 	data := []byte("hello world")
-	msg := message{
+	msg := Message{
 		Topic: "topic a",
 		Data:  data,
 	}
diff --git a/subscriber.go b/subscriber.go
index a9dd054..9d36c0e 100644
--- a/subscriber.go
+++ b/subscriber.go
@@ -1,4 +1,4 @@
-package main
+package messagebroker
 
 import (
 	"fmt"
diff --git a/topic.go b/topic.go
index af11ac0..32e27a6 100644
--- a/topic.go
+++ b/topic.go
@@ -1,4 +1,4 @@
-package main
+package messagebroker
 
 import (
 	"log/slog"
@@ -35,7 +35,7 @@ func (t *topic) removeSubscriber(addr net.Addr) {
 	delete(t.subscriptions, addr)
 }
 
-func (t *topic) sendMessageToSubscribers(msg message) {
+func (t *topic) sendMessageToSubscribers(msg Message) {
 	t.mu.Lock()
 	subscribers := t.subscriptions
 	t.mu.Unlock()