Improve docs

Author: backkem

README.md

@@ -1,13 +1,16 @@
 # SPAKE2
 
-[![GoDoc](https://godoc.org/github.com/backkem/spake2?status.svg)](https://godoc.org/github.com/backkem/spake2)
+[![GoDoc](https://godoc.org/github.com/backkem/spake2-go?status.svg)](https://godoc.org/github.com/backkem/spake2-go)
+[![Build Status](https://github.com/backkem/spake2-go/actions/workflows/ci.yml/badge.svg)](https://github.com/backkem/spake2-go/actions/workflows/ci.yml)
 
 A Go implementation of the SPAKE2 password-authenticated key exchange protocol.
 
 ## Overview
 
 SPAKE2 is a secure, password-authenticated key exchange protocol that allows two parties to establish a shared secret key based on a low-entropy password without revealing the password to an eavesdropper. This implementation provides both client and server components for the SPAKE2 protocol.
 
+⚠️ **Security Warning**: While this implementation has been tested against the official RFC 9382 test vectors for correctness, it has not undergone formal security review or audit for vulnerabilities. Use at your own risk in production environments. Consider having the code reviewed by security experts before deploying in security-critical applications.
+
 ## Features
 
 - Implements the SPAKE2 protocol as described in [RFC 9382](https://www.rfc-editor.org/rfc/rfc9382).
@@ -79,6 +82,36 @@ func main() {
 }
 ```
 
+## Protocol Flow
+
+Below is a simplified flow of the SPAKE2 protocol exchange between client and server:
+
+```mermaid
+sequenceDiagram
+    participant C as Client
+    participant S as Server
+    C->>S: Start (Client Message)
+    S->>C: Exchange (Server Message)
+    C->>S: Finish (Client Confirm)
+    S->>C: Confirm (Server Confirm)
+    C->>C: Verify
+    Note over C,S: Shared Key Established
+```
+
+_Note: Render this diagram on GitHub or convert to an image for better visibility._
+
+Alternatively, as a simple ASCII representation:
+
+```
+Client               Server
+  |  Start Msg ------> |
+  | <------ Exchange Msg |
+  |  Finish Msg ------> |
+  | <------ Confirm Msg  |
+  | Verify             |
+  v Shared Key         v Shared Key
+```
+
 ## Documentation
 
 Full documentation is available on [GoDoc](https://godoc.org/github.com/backkem/spake2).