diff --git a/cmd/uftp/main.go b/cmd/uftp/main.go
index 5f7992f..433f403 100644
--- a/cmd/uftp/main.go
+++ b/cmd/uftp/main.go
@@ -13,6 +13,10 @@ func main() {
 		if err != nil {
 			panic(err)
 		}
+		err = server.SavePublicKeyPem()
+		if err != nil {
+			panic(err)
+		}
 		server.Serve()
 	} else {
 		client.GetFile(os.Args[2], os.Args[3])
diff --git a/protocol.md b/protocol.md
index 089a15c..c966275 100644
--- a/protocol.md
+++ b/protocol.md
@@ -21,7 +21,7 @@
 
 ### Packet Structure
 37 Byte Secure Header | 5 Byte Packet Header | <= 446 Byte Data | 16 Byte chacha20poly1305 overhead
-
+                      | ---------------ENCRYPTED--------------- |
 
 ### Data