diff --git a/.changeset/true-actors-grow.md b/.changeset/true-actors-grow.md
new file mode 100644
index 000000000..b6c43315e
--- /dev/null
+++ b/.changeset/true-actors-grow.md
@@ -0,0 +1,5 @@
+---
+"@turnkey/iframe-stamper": minor
+---
+
+Add optional address parameter for methods intended to be used within the export-and-sign iframe. Also improves documentation (TypeDocs)
diff --git a/examples/wallet-export-sign/README.md b/examples/wallet-export-sign/README.md
index f357624f4..85de79b30 100644
--- a/examples/wallet-export-sign/README.md
+++ b/examples/wallet-export-sign/README.md
@@ -11,6 +11,16 @@ This example includes API stubs to make `whoami` requests and export wallets and
     <img src="./img/demo.png" width="2000"/>
 </p>
 
+### DevEx Summary
+
+1. Get iframe public key from export-and-sign iframe (located at export-and-sign.turnkey.com) via `iframeStamper.publicKey()` OR `await iframeStamper.getEmbeddedPublicKey()`. Generally, the latter is recommended. If it does not exist, you can initialize one via `await iframeStamper.initEmbeddedKey()`.
+2. Use iframe public key in any `export` activity (e.g. `exportWalletAccount`).
+3. Inject resulting bundle from (2) into the iframe via `await iframeStamper.injectKeyExportBundle()`. Note that the `address` param is optional, though recommended in case you would like multiple keys to live in the iframe at a time.
+
+- Note that this step wipes out the iframe's embedded key (as we only want to use it to decrypt one export bundle at most). This means that for subsequent `export` activities, you will need to re-initialize the embedded key via `await iframeStamper.initEmbeddedKey()`.
+
+4. Now, the iframe contains the key material you need to start performing signing. Note that this is stored strictly in-memory. Any page reload would wipe out this state, at which point we recommend starting from step (1).
+
 ## Getting started
 
 ### 1/ Cloning the example
diff --git a/examples/wallet-export-sign/src/components/Export.module.css b/examples/wallet-export-sign/src/components/Export.module.css
new file mode 100644
index 000000000..4b9c19602
--- /dev/null
+++ b/examples/wallet-export-sign/src/components/Export.module.css
@@ -0,0 +1,269 @@
+.container {
+  margin-top: 32px;
+  display: flex;
+  flex-direction: column;
+  gap: 28px;
+  width: 100%;
+  max-width: 100%;
+  font-family:
+    Inter,
+    system-ui,
+    -apple-system,
+    "Segoe UI",
+    Roboto,
+    "Helvetica Neue",
+    Arial,
+    sans-serif;
+  padding: 0;
+  box-sizing: border-box;
+}
+
+.card {
+  padding: 0;
+  border-radius: 12px;
+  border: 1px solid #e5e7eb;
+  background: #ffffff;
+  box-shadow: 0 1px 2px rgba(0, 0, 0, 0.04);
+  transition: all 0.2s ease;
+  width: 100%;
+  box-sizing: border-box;
+  overflow: hidden;
+}
+
+.card:hover {
+  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.06);
+  border-color: #d1d5db;
+}
+
+.cardHeader {
+  display: flex;
+  justify-content: space-between;
+  align-items: center;
+  padding: 20px 24px;
+  background: linear-gradient(to bottom, #fafbfc, #f8f9fa);
+  border-bottom: 1px solid #e5e7eb;
+  gap: 16px;
+}
+
+.cardTitle {
+  font-size: 15px;
+  font-weight: 600;
+  color: #111827;
+  margin: 0;
+  letter-spacing: -0.01em;
+}
+
+.cardSubtitle {
+  font-size: 12px;
+  color: #6b7280;
+  font-weight: 500;
+  white-space: nowrap;
+  text-transform: uppercase;
+  letter-spacing: 0.03em;
+}
+
+.cardBody {
+  padding: 20px;
+}
+
+.textarea {
+  width: 100%;
+  min-height: 110px;
+  padding: 14px 16px;
+  font-family:
+    ui-monospace, SFMono-Regular, "SF Mono", Menlo, Consolas, "Liberation Mono",
+    monospace;
+  font-size: 14px;
+  line-height: 1.6;
+  border: 1px solid #d1d5db;
+  border-radius: 8px;
+  resize: none;
+  transition:
+    border-color 0.2s ease,
+    box-shadow 0.2s ease,
+    background-color 0.2s ease;
+  background: #fafafa;
+  box-sizing: border-box;
+}
+
+.textarea:focus {
+  outline: none;
+  border-color: #2563eb;
+  box-shadow: 0 0 0 3px rgba(37, 99, 235, 0.08);
+  background: #ffffff;
+}
+
+.textarea::placeholder {
+  color: #9ca3af;
+}
+
+.buttonGroup {
+  display: flex;
+  gap: 12px;
+  margin-top: 16px;
+  flex-wrap: wrap;
+}
+
+.button {
+  padding: 11px 20px;
+  font-size: 14px;
+  font-weight: 500;
+  border-radius: 8px;
+  border: 1px solid #d1d5db;
+  background: #ffffff;
+  color: #374151;
+  cursor: pointer;
+  transition: all 0.15s ease;
+  font-family: Inter, system-ui, sans-serif;
+  letter-spacing: -0.01em;
+}
+
+.button:hover:not(:disabled) {
+  background: #f9fafb;
+  border-color: #9ca3af;
+}
+
+.button:active:not(:disabled) {
+  transform: translateY(1px);
+}
+
+.button:disabled {
+  opacity: 0.5;
+  cursor: not-allowed;
+}
+
+.buttonPrimary {
+  background: #050a0b;
+  color: #ffffff;
+  border-color: #050a0b;
+}
+
+.buttonPrimary:hover:not(:disabled) {
+  background: #1a1f23;
+  border-color: #1a1f23;
+}
+
+.outputSection {
+  margin-top: 24px;
+  padding: 20px;
+  background: #f9fafb;
+  border-radius: 8px;
+  border: 1px solid #e5e7eb;
+}
+
+.outputLabel {
+  font-size: 12px;
+  font-weight: 600;
+  color: #6b7280;
+  margin-bottom: 12px;
+  letter-spacing: 0.03em;
+  text-transform: uppercase;
+}
+
+.monoBox {
+  font-family:
+    ui-monospace, SFMono-Regular, "SF Mono", Menlo, Consolas, "Liberation Mono",
+    monospace;
+  font-size: 12px;
+  padding: 14px 16px;
+  border-radius: 6px;
+  background: #ffffff;
+  border: 1px solid #e5e7eb;
+  word-break: break-all;
+  white-space: pre-wrap;
+  color: #1f2937;
+  line-height: 1.7;
+  max-height: 140px;
+  overflow-y: auto;
+  box-sizing: border-box;
+}
+
+.monoBox::-webkit-scrollbar {
+  width: 8px;
+  height: 8px;
+}
+
+.monoBox::-webkit-scrollbar-track {
+  background: #f1f3f5;
+  border-radius: 4px;
+}
+
+.monoBox::-webkit-scrollbar-thumb {
+  background: #cbd5e0;
+  border-radius: 4px;
+}
+
+.monoBox::-webkit-scrollbar-thumb:hover {
+  background: #a0aec0;
+}
+
+.copyButton {
+  padding: 8px 14px;
+  font-size: 12px;
+  font-weight: 500;
+  border-radius: 6px;
+  border: 1px solid #d1d5db;
+  background: #ffffff;
+  color: #374151;
+  cursor: pointer;
+  transition: all 0.15s ease;
+  margin-top: 12px;
+  letter-spacing: -0.01em;
+}
+
+.copyButton:hover {
+  background: #f9fafb;
+  border-color: #9ca3af;
+}
+
+.copyButton:active {
+  transform: translateY(1px);
+}
+
+.iframeContainer {
+  margin-bottom: 20px;
+}
+
+@media (min-width: 640px) {
+  .cardBody {
+    padding: 24px;
+  }
+
+  .textarea {
+    min-height: 120px;
+  }
+}
+
+@media (min-width: 768px) {
+  .cardHeader {
+    padding: 24px 28px;
+  }
+
+  .cardBody {
+    padding: 28px;
+  }
+
+  .cardTitle {
+    font-size: 16px;
+  }
+
+  .textarea {
+    font-size: 14px;
+    padding: 16px 18px;
+    min-height: 130px;
+  }
+
+  .container {
+    gap: 32px;
+  }
+}
+
+@media (min-width: 1024px) {
+  .cardHeader {
+    padding: 24px 32px;
+  }
+
+  .cardBody {
+    padding: 32px;
+  }
+}
diff --git a/examples/wallet-export-sign/src/components/Export.tsx b/examples/wallet-export-sign/src/components/Export.tsx
index 0e865b8ba..e7fbb0c7e 100644
--- a/examples/wallet-export-sign/src/components/Export.tsx
+++ b/examples/wallet-export-sign/src/components/Export.tsx
@@ -6,6 +6,7 @@ import {
   MessageType,
 } from "@turnkey/iframe-stamper";
 import { Dispatch, SetStateAction, useEffect, useState } from "react";
+import styles from "./Export.module.css";
 
 interface ExportProps {
   iframeUrl: string;
@@ -13,50 +14,9 @@ interface ExportProps {
   iframeDisplay: string;
   setIframeStamper: Dispatch<SetStateAction<IframeStamper | null>>;
   showSigning?: boolean; // Only show signing UI for wallet accounts, not wallets
+  walletAccountAddress?: string; // Address of the wallet account being exported
 }
 
-const containerStyles: React.CSSProperties = {
-  marginTop: 16,
-  display: "flex",
-  flexDirection: "column",
-  gap: 16,
-  width: "100%",
-  maxWidth: "100%",
-  fontFamily:
-    "Inter, system-ui, -apple-system, 'Segoe UI', Roboto, 'Helvetica Neue', Arial",
-  maxHeight: "min(70vh, 600px)",
-  overflowY: "auto",
-  overflowX: "hidden",
-  padding: "0",
-  boxSizing: "border-box",
-};
-
-const cardStyles: React.CSSProperties = {
-  padding: 16,
-  borderRadius: 12,
-  border: "1px solid rgba(216,219,227,1)",
-  background: "#ffffff",
-  boxShadow: "0 1px 3px rgba(0,0,0,0.06), 0 1px 2px rgba(0,0,0,0.04)",
-  transition: "box-shadow 0.2s ease",
-  width: "100%",
-  boxSizing: "border-box",
-  overflowX: "hidden",
-};
-
-const monoBox: React.CSSProperties = {
-  fontFamily: "monospace",
-  fontSize: 13,
-  padding: 12,
-  borderRadius: 8,
-  background: "#f8f9fa",
-  border: "1px solid #e0e3e7",
-  wordBreak: "break-word",
-  whiteSpace: "pre-wrap",
-  color: "#2b2f33",
-  maxHeight: "200px",
-  overflowY: "auto",
-};
-
 const iframeCss = `
   iframe {
     box-sizing: border-box;
@@ -66,16 +26,16 @@ const iframeCss = `
     border-radius: 12px;
     border-width: 1px;
     border-style: solid;
-    border-color: rgba(216, 219, 227, 1);
-    padding: 16px;
+    border-color: #e5e7eb;
+    padding: 20px;
     background: white;
-    box-shadow: 0 1px 3px rgba(0,0,0,0.06), 0 1px 2px rgba(0,0,0,0.04);
-    margin-bottom: 8px;
+    box-shadow: 0 1px 3px rgba(0, 0, 0, 0.05);
+    margin-bottom: 0;
   }
 
   @media (min-width: 768px) {
     iframe {
-      padding: 20px;
+      padding: 24px;
     }
   }
 `;
@@ -97,7 +57,7 @@ export function Export(props: ExportProps) {
   const [txSerialized, setTxSerialized] = useState<string>("");
   const [txSigned, setTxSigned] = useState<string>("");
 
-  const [initializing, setInitializing] = useState<boolean>(false);
+  const [_initializing, setInitializing] = useState<boolean>(false);
 
   useEffect(() => {
     setIframeDisplay(props.iframeDisplay);
@@ -146,32 +106,30 @@ export function Export(props: ExportProps) {
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [props.setIframeStamper, iframeStamper]);
 
-  // For the sake of demonstration, only allow signing one message at a time (embedded private key will be cleared after signature)
-  const signMessage = () => {
+  const signMessage = async () => {
     if (iframeStamper === null) {
       alert("Iframe not ready — reveal private key first.");
       return;
     }
 
-    iframeStamper
-      .signMessage({ message, type: MessageType.Solana })
-      .then((sig: string) => {
-        setSignature(sig);
-      })
-      .catch((error: Error) => {
-        console.error("Error signing message:", error);
-        alert("Error signing message: " + error.message);
-      });
-
-    // now clear the embedded private key from iframe's memory
-    iframeStamper.clearEmbeddedPrivateKey().catch((error: Error) => {
-      console.error("Error clearing embedded private key:", error);
-      alert("Error clearing embedded private key: " + error.message);
-    });
+    // At this point, we're relying on having the decrypted (Solana) private key in-memory within the iframe for signing
+    // Note that the embedded key has been wiped out at this point as it was used once to initially decrypt.
+    try {
+      const signedMessage = await iframeStamper.signMessage(
+        {
+          message,
+          type: MessageType.Solana,
+        },
+        props.walletAccountAddress,
+      );
+      setSignature(signedMessage);
+    } catch (error: any) {
+      console.error("Error signing message:", error);
+      alert("Error signing message: " + error.message);
+    }
   };
 
-  // For the sake of demonstration, only allow signing one message at a time (embedded private key will be cleared after signature)
-  const signTransaction = () => {
+  const signTransaction = async () => {
     if (iframeStamper === null) {
       alert("Iframe not ready — reveal private key first.");
       return;
@@ -182,24 +140,22 @@ export function Export(props: ExportProps) {
       return;
     }
 
-    iframeStamper
-      .signTransaction({
-        transaction: txSerialized,
-        type: TransactionType.Solana,
-      })
-      .then((signed: string) => {
-        setTxSigned(signed);
-      })
-      .catch((error: Error) => {
-        console.error("Error signing transaction:", error);
-        alert("Error signing transaction: " + error.message);
-      });
-
-    // now clear the embedded private key from iframe's memory
-    iframeStamper.clearEmbeddedPrivateKey().catch((error: Error) => {
-      console.error("Error clearing embedded private key:", error);
-      alert("Error clearing embedded private key: " + error.message);
-    });
+    // At this point, we're relying on having the decrypted (Solana) private key in-memory within the iframe for signing
+    // Note that the embedded key has been wiped out at this point as it was used once to initially decrypt.
+    try {
+      const signedTransaction = await iframeStamper.signTransaction(
+        {
+          transaction: txSerialized,
+          type: TransactionType.Solana,
+        },
+        props.walletAccountAddress,
+      );
+
+      setTxSigned(signedTransaction);
+    } catch (error: any) {
+      console.error("Error signing transaction:", error);
+      alert("Error signing transaction: " + error.message);
+    }
   };
 
   const copyToClipboard = async (text: string) => {
@@ -216,145 +172,116 @@ export function Export(props: ExportProps) {
   return (
     <div>
       {/* keep iframe mounted but visually controlled by parent */}
-      <div style={{ display: iframeDisplay }} id={TurnkeyIframeContainerId}>
+      <div
+        className={styles.iframeContainer}
+        style={{ display: iframeDisplay }}
+        id={TurnkeyIframeContainerId}
+      >
         <style>{iframeCss}</style>
       </div>
 
       {iframeDisplay === "block" && props.showSigning ? (
-        <div style={containerStyles}>
+        <div className={styles.container}>
           {/* Message signing */}
-          <div style={cardStyles}>
-            <div
-              style={{
-                display: "flex",
-                justifyContent: "space-between",
-                alignItems: "center",
-              }}
-            >
-              <strong>Sign arbitrary message</strong>
-              <small style={{ color: "#666" }}>ed25519 signature</small>
+          <div className={styles.card}>
+            <div className={styles.cardHeader}>
+              <h3 className={styles.cardTitle}>Sign arbitrary message</h3>
+              <span className={styles.cardSubtitle}>ed25519</span>
             </div>
 
-            <textarea
-              value={message}
-              onChange={(e) => setMessage(e.target.value)}
-              style={{
-                width: "100%",
-                height: 88,
-                marginTop: 10,
-                padding: 10,
-                fontFamily: "monospace",
-                fontSize: 13,
-              }}
-            />
-
-            <div style={{ display: "flex", gap: 8, marginTop: 10 }}>
-              <button
-                onClick={signMessage}
-                disabled={!iframeStamper}
-                style={{
-                  padding: "8px 12px",
-                  cursor: iframeStamper ? "pointer" : "not-allowed",
-                }}
-              >
-                Sign message
-              </button>
-
-              <button
-                onClick={() => {
-                  setMessage("");
-                  setSignature("");
-                }}
-                style={{ padding: "8px 12px" }}
-              >
-                Clear
-              </button>
-            </div>
+            <div className={styles.cardBody}>
+              <textarea
+                className={styles.textarea}
+                value={message}
+                onChange={(e) => setMessage(e.target.value)}
+              />
+
+              <div className={styles.buttonGroup}>
+                <button
+                  className={`${styles.button} ${styles.buttonPrimary}`}
+                  onClick={signMessage}
+                  disabled={!iframeStamper}
+                >
+                  Sign message
+                </button>
+
+                <button
+                  className={styles.button}
+                  onClick={() => {
+                    setMessage("");
+                    setSignature("");
+                  }}
+                >
+                  Clear
+                </button>
+              </div>
 
-            {signature ? (
-              <div style={{ marginTop: 12 }}>
-                <div style={{ fontSize: 13, color: "#444" }}>
-                  Signature (hex)
-                </div>
-                <div style={{ ...monoBox, marginTop: 8 }}>{signature}</div>
-                <div style={{ display: "flex", gap: 8, marginTop: 8 }}>
+              {signature ? (
+                <div className={styles.outputSection}>
+                  <div className={styles.outputLabel}>Signature (hex)</div>
+                  <div className={styles.monoBox}>{signature}</div>
                   <button
+                    className={styles.copyButton}
                     onClick={() => copyToClipboard(signature)}
-                    style={{ padding: "6px 10px" }}
                   >
                     Copy signature
                   </button>
                 </div>
-              </div>
-            ) : null}
+              ) : null}
+            </div>
           </div>
 
           {/* Transaction signing */}
-          <div style={cardStyles}>
-            <div
-              style={{
-                display: "flex",
-                justifyContent: "space-between",
-                alignItems: "center",
-              }}
-            >
-              <strong>Sign Solana transaction</strong>
-              <small style={{ color: "#666" }}>Hex-encoded transaction</small>
+          <div className={styles.card}>
+            <div className={styles.cardHeader}>
+              <h3 className={styles.cardTitle}>Sign Solana transaction</h3>
+              <span className={styles.cardSubtitle}>Hex-encoded</span>
             </div>
 
-            <textarea
-              value={txSerialized}
-              onChange={(e) => setTxSerialized(e.target.value)}
-              placeholder="Paste hex-encoded transaction here"
-              style={{
-                width: "100%",
-                height: 120,
-                marginTop: 10,
-                padding: 10,
-                fontFamily: "monospace",
-                fontSize: 13,
-              }}
-            />
-
-            <div style={{ display: "flex", gap: 8, marginTop: 10 }}>
-              <button
-                onClick={signTransaction}
-                disabled={!iframeStamper}
-                style={{
-                  padding: "8px 12px",
-                  cursor: iframeStamper ? "pointer" : "not-allowed",
-                }}
-              >
-                Sign transaction
-              </button>
-
-              <button
-                onClick={() => {
-                  setTxSerialized("");
-                  setTxSigned("");
-                }}
-                style={{ padding: "8px 12px" }}
-              >
-                Clear
-              </button>
-            </div>
+            <div className={styles.cardBody}>
+              <textarea
+                className={styles.textarea}
+                value={txSerialized}
+                onChange={(e) => setTxSerialized(e.target.value)}
+                placeholder="Paste hex-encoded transaction here"
+                style={{ minHeight: "120px" }}
+              />
+
+              <div className={styles.buttonGroup}>
+                <button
+                  className={`${styles.button} ${styles.buttonPrimary}`}
+                  onClick={signTransaction}
+                  disabled={!iframeStamper}
+                >
+                  Sign transaction
+                </button>
+
+                <button
+                  className={styles.button}
+                  onClick={() => {
+                    setTxSerialized("");
+                    setTxSigned("");
+                  }}
+                >
+                  Clear
+                </button>
+              </div>
 
-            {txSigned ? (
-              <div style={{ marginTop: 12 }}>
-                <div style={{ fontSize: 13, color: "#444" }}>
-                  Signed transaction (hex)
-                </div>
-                <div style={{ ...monoBox, marginTop: 8 }}>{txSigned}</div>
-                <div style={{ display: "flex", gap: 8, marginTop: 8 }}>
+              {txSigned ? (
+                <div className={styles.outputSection}>
+                  <div className={styles.outputLabel}>
+                    Signed transaction (hex)
+                  </div>
+                  <div className={styles.monoBox}>{txSigned}</div>
                   <button
+                    className={styles.copyButton}
                     onClick={() => copyToClipboard(txSigned)}
-                    style={{ padding: "6px 10px" }}
                   >
                     Copy signed tx
                   </button>
                 </div>
-              </div>
-            ) : null}
+              ) : null}
+            </div>
           </div>
         </div>
       ) : null}
diff --git a/examples/wallet-export-sign/src/components/ExportWalletAccount.tsx b/examples/wallet-export-sign/src/components/ExportWalletAccount.tsx
index d024341bc..f267e2024 100644
--- a/examples/wallet-export-sign/src/components/ExportWalletAccount.tsx
+++ b/examples/wallet-export-sign/src/components/ExportWalletAccount.tsx
@@ -31,15 +31,27 @@ export function ExportWalletAccount(props: ExportWalletAccountProps) {
       return;
     }
 
+    // Check to ensure there's an embedded key
+    let embeddedKey = await iframeStamper.getEmbeddedPublicKey();
+
+    if (!embeddedKey) {
+      console.warn(
+        "Iframe not ready — embedded key not found. Creating a new one...",
+      );
+
+      embeddedKey = await iframeStamper.initEmbeddedKey();
+    }
+
     const response = await axios.post("/api/exportWalletAccount", {
       walletAccountAddress: props.walletAccountAddress,
-      targetPublicKey: iframeStamper.publicKey(),
+      targetPublicKey: embeddedKey,
     });
 
     let injected = await iframeStamper.injectKeyExportBundle(
       response.data["exportBundle"],
       props.organizationId,
       KeyFormat.Hexadecimal,
+      props.walletAccountAddress,
     );
     if (injected !== true) {
       alert("Unexpected error while injecting export bundle.");
@@ -110,6 +122,7 @@ export function ExportWalletAccount(props: ExportWalletAccountProps) {
           iframeUrl={process.env.NEXT_PUBLIC_EXPORT_SIGN_IFRAME_URL!}
           turnkeyBaseUrl={process.env.NEXT_PUBLIC_BASE_URL!}
           showSigning={props.addressFormat === "ADDRESS_FORMAT_SOLANA"}
+          walletAccountAddress={props.walletAccountAddress}
         />
       </div>
     </div>
diff --git a/examples/wallet-export-sign/src/components/Modal.tsx b/examples/wallet-export-sign/src/components/Modal.tsx
index be8e5ae39..413568be6 100644
--- a/examples/wallet-export-sign/src/components/Modal.tsx
+++ b/examples/wallet-export-sign/src/components/Modal.tsx
@@ -33,21 +33,23 @@ export function Modal(props: ModalProps) {
   return (
     <div className={styles.modalOverlay}>
       <div className={styles.customModal}>
-        <button
-          type="button"
-          onClick={props.onClose}
-          className={styles.closeButton}
-        >
-          <span className={styles.srOnly}>Close menu</span>
-          <Image
-            className={`inline-block`}
-            src="/close.svg"
-            alt="Close"
-            width={20}
-            height={20}
-            priority
-          />
-        </button>
+        <div className={styles.modalHeader}>
+          <button
+            type="button"
+            onClick={props.onClose}
+            className={styles.closeButton}
+          >
+            <span className={styles.srOnly}>Close menu</span>
+            <Image
+              className={`inline-block`}
+              src="/close.svg"
+              alt="Close"
+              width={20}
+              height={20}
+              priority
+            />
+          </button>
+        </div>
         {props.children}
       </div>
     </div>
diff --git a/examples/wallet-export-sign/src/pages/index.module.css b/examples/wallet-export-sign/src/pages/index.module.css
index 69688b3c4..2bcf6c86d 100644
--- a/examples/wallet-export-sign/src/pages/index.module.css
+++ b/examples/wallet-export-sign/src/pages/index.module.css
@@ -19,13 +19,33 @@
 .main {
   display: flex;
   flex-direction: column;
-  justify-content: space-between;
+  justify-content: flex-start;
   align-items: center;
-  padding: 6rem;
+  padding: 2.5rem 1.5rem;
+  min-height: 100vh;
+  background: linear-gradient(to bottom, #fafbfc 0%, #ffffff 100%);
+}
+
+@media (min-width: 768px) {
+  .main {
+    padding: 4rem 2.5rem;
+  }
+}
+
+@media (min-width: 1024px) {
+  .main {
+    padding: 5rem 3rem;
+  }
 }
 
 .header {
-  margin-bottom: 50px;
+  margin-bottom: 2.5rem;
+}
+
+@media (min-width: 768px) {
+  .header {
+    margin-bottom: 3rem;
+  }
 }
 
 .table {
@@ -33,21 +53,26 @@
   width: 100%;
   font-family: "Inter";
   margin-top: 10px;
+  background: #ffffff;
+  border-radius: 12px;
+  overflow: hidden;
+  box-shadow: 0 1px 3px rgba(0, 0, 0, 0.05);
 }
 
 .tableHeader {
-  background-color: #f9fbff;
-  border: 1px solid #ebedf2;
+  background: linear-gradient(to bottom, #fafbfc, #f5f7fa);
+  border: 1px solid #e5e7eb;
   margin: 0;
 }
 
 .tableRow {
-  border: 1px solid #ebedf2;
+  border: 1px solid #e5e7eb;
   border-top: none;
+  transition: all 0.15s ease;
 }
 
 .tableRow:hover {
-  background-color: #f9fbff;
+  background-color: #f9fafb;
 }
 
 .tableRowClickable {
@@ -55,31 +80,28 @@
 }
 
 .tableHeaderCell {
-  padding: 14px;
-  font-weight: 500;
-  font-size: 0.875rem;
-  line-height: 1.25rem;
-  color: #6c727e;
+  padding: 18px 20px;
+  font-weight: 600;
+  font-size: 0.75rem;
+  line-height: 1rem;
+  color: #6b7280;
   text-align: left;
   align-items: center;
+  text-transform: uppercase;
+  letter-spacing: 0.05em;
 }
 
 .cell {
-  padding: 14px;
+  padding: 20px;
 }
 
 .cell p {
-  color: #555b64;
+  color: #374151;
   font-weight: 400;
-  font-size: 0.875rem;
-  line-height: 1.25rem;
+  font-size: 0.9375rem;
+  line-height: 1.5rem;
   margin: 0;
   padding: 0;
-  display: block;
-  margin-block-start: 1em;
-  margin-block-end: 1em;
-  margin-inline-start: 0px;
-  margin-inline-end: 0px;
 }
 
 .noWallets {
@@ -104,8 +126,10 @@
   gap: 12px;
   justify-content: baseline;
   font-family:
-    "iA Writer Mono", ui-monospace, SFMono-Regular, "SF Mono", Menlo, Consolas,
-    "Liberation Mono", monospace;
+    ui-monospace, SFMono-Regular, "SF Mono", Menlo, Consolas, "Liberation Mono",
+    monospace;
+  font-size: 0.875rem;
+  color: #374151;
 }
 
 .buttons {
@@ -131,18 +155,24 @@
   display: flex;
   align-items: center;
   justify-content: center;
-  background-color: #f1f3f6;
-  border: none;
-  border-radius: 50%;
-  width: 25px;
-  height: 25px;
+  background-color: #f3f4f6;
+  border: 1px solid #e5e7eb;
+  border-radius: 6px;
+  width: 32px;
+  height: 32px;
   cursor: pointer;
   outline: none;
-  transition: background-color 0.2s;
+  transition: all 0.15s ease;
 }
 
 .exportButton:hover {
-  background-color: #ebedf2;
+  background-color: #e5e7eb;
+  border-color: #d1d5db;
+  transform: translateY(-1px);
+}
+
+.exportButton:active {
+  transform: translateY(0);
 }
 
 .modalButton,
@@ -150,28 +180,36 @@
   background-color: #050a0b;
   border: none;
   border-radius: 8px;
-  height: 44px;
+  height: 46px;
   cursor: pointer;
   outline: none;
   font-family: "Inter";
   border: 1px solid #050a0b;
-  margin-bottom: 20px;
+  margin-bottom: 16px;
   color: #fff;
   font-size: 15px;
-  font-weight: 500;
-  transition: all 0.2s ease;
+  font-weight: 600;
+  transition: all 0.15s ease;
+  letter-spacing: -0.01em;
 }
 
 .modalButton:hover,
 .longModalButton:hover {
   background-color: #1a1f23;
-  transform: translateY(-1px);
-  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+  transform: translateY(-2px);
+  box-shadow: 0 4px 8px rgba(0, 0, 0, 0.15);
+}
+
+.modalButton:active,
+.longModalButton:active {
+  transform: translateY(0);
+  box-shadow: 0 1px 2px rgba(0, 0, 0, 0.1);
 }
 
 .modalButton {
   width: 100%;
   max-width: 200px;
+  padding: 0 24px;
 }
 
 @media (min-width: 768px) {
@@ -191,51 +229,73 @@
   display: flex;
   justify-content: center;
   width: 100%;
-  padding: 20px 0;
+  padding: 28px 0 0;
 }
 
 .modalDetails {
   display: flex;
-  justify-content: center;
+  justify-content: flex-start;
   flex-direction: column;
   align-items: center;
-  padding: 10px;
+  padding: 24px 20px 32px;
   width: 100%;
   box-sizing: border-box;
   overflow-x: hidden;
 }
 
+@media (min-width: 640px) {
+  .modalDetails {
+    padding: 28px 28px 36px;
+  }
+}
+
 @media (min-width: 768px) {
   .modalDetails {
-    padding: 20px;
+    padding: 32px 40px 44px;
+  }
+}
+
+@media (min-width: 1024px) {
+  .modalDetails {
+    padding: 36px 48px 48px;
   }
 }
 
 .modalDetails h2 {
   text-align: center;
   font-family: "Inter";
-  padding-top: 20px;
-  font-size: 24px;
-  margin-bottom: 16px;
+  padding-top: 0;
+  font-size: 22px;
+  margin-bottom: 20px;
+  font-weight: 700;
+  color: #111827;
+  letter-spacing: -0.02em;
+  line-height: 1.3;
 }
 
 @media (min-width: 768px) {
   .modalDetails h2 {
-    padding-top: 40px;
-    font-size: 28px;
+    padding-top: 0;
+    font-size: 24px;
+    margin-bottom: 24px;
   }
 }
 
 .modalDetails ul {
   margin-top: 0;
   text-align: left;
+  padding-left: 24px;
+}
+
+.modalDetails li {
+  margin-bottom: 16px;
 }
 
 .modalDetails p {
   font-family: "Inter";
   font-size: 14px;
-  color: #555b64;
-  line-height: 1.6;
+  color: #4b5563;
+  line-height: 1.65;
   word-wrap: break-word;
   overflow-wrap: break-word;
   max-width: 100%;
@@ -244,6 +304,7 @@
 @media (min-width: 768px) {
   .modalDetails p {
     font-size: 15px;
+    line-height: 1.7;
   }
 }
 
@@ -262,50 +323,82 @@
 
 .customModal {
   background-color: rgba(255, 255, 255, 1);
-  padding: 1.5rem;
-  border-radius: 0.75rem;
+  padding: 0;
+  border-radius: 16px;
   box-shadow:
-    0 10px 15px -3px rgba(0, 0, 0, 0.1),
-    0 4px 6px -2px rgba(0, 0, 0, 0.05);
-  width: 90%;
-  max-width: 700px;
-  max-height: 90vh;
-  overflow-y: auto;
-  overflow-x: hidden;
+    0 20px 25px -5px rgba(0, 0, 0, 0.1),
+    0 10px 10px -5px rgba(0, 0, 0, 0.04);
+  width: 94%;
+  max-width: 520px;
+  max-height: 92vh;
+  overflow: hidden;
   box-sizing: border-box;
+  border: 1px solid rgba(229, 231, 235, 0.8);
+  display: flex;
+  flex-direction: column;
+}
+
+@media (min-width: 640px) {
+  .customModal {
+    max-width: 600px;
+  }
 }
 
 @media (min-width: 768px) {
   .customModal {
-    width: 80%;
+    width: 90%;
+    max-width: 700px;
   }
 }
 
 @media (min-width: 1024px) {
   .customModal {
-    width: 60%;
-    max-width: 700px;
+    width: 80%;
+    max-width: 800px;
+  }
+}
+
+@media (min-width: 1280px) {
+  .customModal {
+    max-width: 900px;
   }
 }
 
 .modalInner {
   width: 100%;
-  min-height: 400px;
   display: flex;
   flex-direction: column;
-  overflow-x: hidden;
   box-sizing: border-box;
+  overflow-y: auto;
+  overflow-x: hidden;
+  flex: 1;
+  min-height: 0;
+}
+
+.modalHeader {
+  display: flex;
+  align-items: center;
+  justify-content: flex-end;
+  padding: 20px 24px 0;
+  flex-shrink: 0;
 }
 
 .closeButton {
-  background-color: #ffffff; /* bg-white */
+  background-color: transparent;
   border: none;
-  padding: 0.5rem; /* p-2 */
-  display: inline-flex; /* inline-flex */
-  align-items: center; /* items-center */
-  justify-content: center; /* justify-center */
-  color: #9ca3af; /* text-gray-400 */
+  padding: 0.5rem;
+  display: inline-flex;
+  align-items: center;
+  justify-content: center;
+  color: #9ca3af;
   cursor: pointer;
+  border-radius: 6px;
+  transition: all 0.15s ease;
+}
+
+.closeButton:hover {
+  background-color: #f3f4f6;
+  color: #6b7280;
 }
 
 .srOnly {
@@ -352,56 +445,62 @@
   display: flex;
   justify-content: center;
   color: #f59e0b;
-  margin-bottom: 20px;
+  margin-bottom: 24px;
 }
 
 .warningIcon svg {
-  width: 40px;
-  height: 40px;
+  width: 44px;
+  height: 44px;
+  filter: drop-shadow(0 2px 4px rgba(245, 158, 11, 0.2));
 }
 
 @media (min-width: 768px) {
   .warningIcon svg {
-    width: 48px;
-    height: 48px;
+    width: 52px;
+    height: 52px;
   }
 }
 
 .warningIntro {
   font-weight: 500;
-  margin-bottom: 16px;
+  margin-bottom: 20px;
   font-size: 14px;
+  line-height: 1.6;
 }
 
 @media (min-width: 768px) {
   .warningIntro {
     font-size: 15px;
+    margin-bottom: 24px;
   }
 }
 
 .warningList {
   text-align: left;
-  padding-left: 20px;
-  padding-right: 10px;
+  padding-left: 24px;
+  padding-right: 12px;
   width: 100%;
   box-sizing: border-box;
   overflow-x: hidden;
+  margin-bottom: 8px;
 }
 
 .warningList li {
-  margin-bottom: 16px;
+  margin-bottom: 18px;
   word-wrap: break-word;
   overflow-wrap: break-word;
+  line-height: 1.65;
 }
 
 @media (min-width: 768px) {
   .warningList {
-    padding-left: 24px;
+    padding-left: 28px;
     padding-right: 0;
   }
 
   .warningList li {
     margin-bottom: 20px;
+    line-height: 1.7;
   }
 }
 
diff --git a/examples/wallet-export-sign/src/pages/wallet/[id].tsx b/examples/wallet-export-sign/src/pages/wallet/[id].tsx
index 325e2944c..86f514d0a 100644
--- a/examples/wallet-export-sign/src/pages/wallet/[id].tsx
+++ b/examples/wallet-export-sign/src/pages/wallet/[id].tsx
@@ -136,7 +136,6 @@ export default function WalletDetailsPage() {
                   className={styles.tableHeaderCell}
                   style={{ width: "55px" }}
                 ></th>
-                <th className={styles.tableHeaderCell}>Account Name</th>
                 <th className={styles.tableHeaderCell}>Address</th>
                 <th className={styles.tableHeaderCell}>Path</th>
               </tr>
@@ -163,9 +162,6 @@ export default function WalletDetailsPage() {
                           />
                         </button>
                       </td>
-                      <td className={styles.cell}>
-                        <p>{account.walletAccountId}</p>
-                      </td>
                       <td className={styles.cell}>
                         <p className={styles.addressCell}>{account.address}</p>
                       </td>
diff --git a/examples/with-solana/package.json b/examples/with-solana/package.json
index dc11889f5..b5300b9f1 100644
--- a/examples/with-solana/package.json
+++ b/examples/with-solana/package.json
@@ -11,6 +11,7 @@
     "jupiter-swap": "pnpm tsx src/jupiterSwap.ts",
     "with-fee-payer": "pnpm tsx src/withFeePayer.ts",
     "add-signature": "pnpm tsx src/addSignature.ts",
+    "broadcast": "pnpm tsx src/broadcast.ts",
     "clean": "rimraf ./dist ./.cache",
     "typecheck": "tsc --noEmit"
   },
diff --git a/examples/with-solana/src/broadcast.ts b/examples/with-solana/src/broadcast.ts
new file mode 100644
index 000000000..eb04a6c30
--- /dev/null
+++ b/examples/with-solana/src/broadcast.ts
@@ -0,0 +1,121 @@
+import * as dotenv from "dotenv";
+import * as path from "path";
+import prompts from "prompts";
+import { Transaction, VersionedTransaction } from "@solana/web3.js";
+
+// Load environment variables from `.env.local`
+dotenv.config({ path: path.resolve(process.cwd(), ".env.local") });
+
+import { solanaNetwork } from "./utils";
+
+async function main() {
+  // Create a node connection; if no env var is found, default to public devnet RPC
+  const nodeEndpoint =
+    process.env.SOLANA_NODE || "https://api.devnet.solana.com";
+  const connection = solanaNetwork.connect(nodeEndpoint);
+
+  console.log(`\nConnected to Solana network: ${nodeEndpoint}\n`);
+
+  // Prompt for the signed transaction payload
+  const { payload } = await prompts([
+    {
+      type: "text",
+      name: "payload",
+      message: "Enter the signed transaction payload (hex or base64):",
+      validate: (value) =>
+        value && value.trim().length > 0 ? true : "Payload cannot be empty",
+    },
+  ]);
+
+  if (!payload) {
+    console.log("No payload provided. Exiting.");
+    process.exit(0);
+  }
+
+  let signedTransaction: Transaction | VersionedTransaction;
+
+  try {
+    // Try to decode as hex first
+    const buffer = Buffer.from(payload.trim(), "hex");
+
+    // Try to deserialize as VersionedTransaction first
+    try {
+      signedTransaction = VersionedTransaction.deserialize(buffer);
+      console.log("\nDetected versioned transaction");
+    } catch {
+      // Fall back to legacy Transaction
+      signedTransaction = Transaction.from(buffer);
+      console.log("\nDetected legacy transaction");
+    }
+  } catch (hexError) {
+    try {
+      // Try base64 if hex fails
+      const buffer = Buffer.from(payload.trim(), "base64");
+
+      try {
+        signedTransaction = VersionedTransaction.deserialize(buffer);
+        console.log("\nDetected versioned transaction (base64)");
+      } catch {
+        signedTransaction = Transaction.from(buffer);
+        console.log("\nDetected legacy transaction (base64)");
+      }
+    } catch (base64Error) {
+      console.error("\nFailed to decode transaction payload.");
+      console.error(
+        "Please provide a valid hex or base64 encoded transaction.",
+      );
+      process.exit(1);
+    }
+  }
+
+  // Verify signatures for legacy transactions
+  if ("verifySignatures" in signedTransaction) {
+    const verified = signedTransaction.verifySignatures();
+    if (!verified) {
+      console.warn(
+        "\n⚠️  Warning: Transaction signatures could not be verified",
+      );
+      const { proceed } = await prompts([
+        {
+          type: "confirm",
+          name: "proceed",
+          message: "Do you want to proceed anyway?",
+          initial: false,
+        },
+      ]);
+
+      if (!proceed) {
+        console.log("Broadcast cancelled.");
+        process.exit(0);
+      }
+    } else {
+      console.log("✓ Transaction signatures verified");
+    }
+  }
+
+  // Confirm before broadcasting
+  const { confirm } = await prompts([
+    {
+      type: "confirm",
+      name: "confirm",
+      message: "Ready to broadcast this transaction?",
+      initial: true,
+    },
+  ]);
+
+  if (!confirm) {
+    console.log("Broadcast cancelled.");
+    process.exit(0);
+  }
+
+  // Broadcast the signed transaction
+  console.log("\nBroadcasting transaction...");
+  await solanaNetwork.broadcast(connection, signedTransaction);
+
+  process.exit(0);
+}
+
+main().catch((error) => {
+  console.error(error);
+  process.exit(1);
+});
diff --git a/packages/iframe-stamper/src/index.ts b/packages/iframe-stamper/src/index.ts
index e8b974b87..596927909 100644
--- a/packages/iframe-stamper/src/index.ts
+++ b/packages/iframe-stamper/src/index.ts
@@ -60,7 +60,7 @@ export enum IframeEventType {
   // Value: MessageChannel port
   TurnkeyInitMessageChannel = "TURNKEY_INIT_MESSAGE_CHANNEL",
   // Event sent by the parent to get the iframe target embedded key's public key.
-  // Value: none
+  // Value: the iframe public key
   GetEmbeddedPublicKey = "GET_EMBEDDED_PUBLIC_KEY",
   // Event sent by the parent to clear the iframe's embedded key.
   // Value: none
@@ -178,6 +178,11 @@ export class IframeStamper {
   /**
    * Creates a new iframe stamper. This function _does not_ insert the iframe in the DOM.
    * Call `.init()` to insert the iframe element in the DOM.
+   * @param {TIframeStamperConfig} config - Configuration object for the iframe stamper
+   * @throws {Error} When running in non-browser environment
+   * @throws {Error} When MessageChannel is not supported
+   * @throws {Error} When iframeContainer is not provided
+   * @throws {Error} When iframe element with the same ID already exists
    */
   constructor(config: TIframeStamperConfig) {
     if (typeof window === "undefined") {
@@ -232,6 +237,11 @@ export class IframeStamper {
     this.pendingRequests = new Map();
   }
 
+  /**
+   * Handles incoming messages from the iframe via MessageChannel
+   * @param {MessageEvent} event - Message event from the iframe
+   * @returns {void}
+   */
   onMessageHandler(event: MessageEvent): void {
     const { type, value, requestId } = event.data || {};
 
@@ -273,7 +283,9 @@ export class IframeStamper {
 
   /**
    * Inserts the iframe on the page and returns a promise resolving to the iframe's public key
-   * @param dangerouslyOverrideIframeKeyTtl Optional TTL override for the iframe's embedded key (default 48 hours). Only use this if you are intentional about the security implications.
+   * @param {number} [dangerouslyOverrideIframeKeyTtl] - Optional TTL override for the iframe's embedded key (default 48 hours). Only use this if you are intentional about the security implications.
+   * @returns {Promise<string>} The iframe's public key
+   * @throws {Error} When contentWindow or contentWindow.postMessage does not exist
    */
   async init(
     dangerouslyOverrideIframeKeyTtl?: number | undefined,
@@ -315,7 +327,8 @@ export class IframeStamper {
   }
 
   /**
-   * Removes the iframe from the DOM
+   * Removes the iframe from the DOM and cleans up all resources
+   * @returns {void}
    */
   clear() {
     this.messageChannel?.port1?.close();
@@ -326,6 +339,7 @@ export class IframeStamper {
 
   /**
    * Returns the public key, or `null` if the underlying iframe isn't properly initialized.
+   * @returns {string | null} The iframe's public key or null
    */
   publicKey(): string | null {
     return this.iframePublicKey;
@@ -334,6 +348,7 @@ export class IframeStamper {
   /**
    * Returns the public key, or `null` if the underlying iframe isn't properly initialized.
    * This differs from the above in that it reaches out to the live iframe to see if an embedded key exists.
+   * @returns {Promise<string | null>} The embedded public key or null
    */
   async getEmbeddedPublicKey(): Promise<string | null> {
     const publicKey = await this.createRequest<string | null>(
@@ -346,6 +361,7 @@ export class IframeStamper {
 
   /**
    * Clears the embedded key within an iframe.
+   * @returns {Promise<null>} Returns null on success
    */
   async clearEmbeddedKey(): Promise<null> {
     await this.createRequest<null>(IframeEventType.ClearEmbeddedKey);
@@ -358,7 +374,7 @@ export class IframeStamper {
    * Creates a new embedded key within an iframe. If an embedded key already exists, this will return it.
    * This is primarily to be used in conjunction with `clearEmbeddedKey()`: after an embedded key is cleared,
    * this can be used to create a new one.
-   * @return {string | null} the newly created embedded public key.
+   * @returns {Promise<string | null>} The newly created embedded public key
    */
   async initEmbeddedKey(): Promise<string | null> {
     const publicKey = await this.createRequest<string | null>(
@@ -371,9 +387,10 @@ export class IframeStamper {
 
   /**
    * Generic function to abstract away request creation
-   * @param type
-   * @param payload
-   * @returns expected shape <T>
+   * @template T
+   * @param {IframeEventType} type - The type of iframe event to send
+   * @param {any} [payload={}] - Optional payload data to send with the request
+   * @returns {Promise<T>} Promise resolving to the expected response shape
    */
   private createRequest<T>(
     type: IframeEventType,
@@ -401,6 +418,8 @@ export class IframeStamper {
    * The bundle should be encrypted to the iframe's initial public key
    * Encryption should be performed with HPKE (RFC 9180).
    * This is used during recovery and auth flows.
+   * @param {string} bundle - The encrypted credential bundle to inject
+   * @returns {Promise<boolean>} Returns true on successful injection
    */
   async injectCredentialBundle(bundle: string): Promise<boolean> {
     return this.createRequest<boolean>(IframeEventType.InjectCredentialBundle, {
@@ -414,16 +433,23 @@ export class IframeStamper {
    * Encryption should be performed with HPKE (RFC 9180).
    * The key format to encode the private key in after it's exported and decrypted: HEXADECIMAL or SOLANA. Defaults to HEXADECIMAL.
    * This is used during the private key export flow.
+   * @param {string} bundle - The encrypted export bundle to inject
+   * @param {string} organizationId - The organization ID
+   * @param {KeyFormat} keyFormat - [Optional] The key format (HEXADECIMAL or SOLANA). Defaults to HEXADECIMAL
+   * @param {string} address - [Optional] Address corresponding to the key bundle (case sensitive)
+   * @returns {Promise<boolean>} Returns true on successful injection
    */
   async injectKeyExportBundle(
     bundle: string,
     organizationId: string,
     keyFormat?: KeyFormat,
+    address?: string,
   ): Promise<boolean> {
     return this.createRequest<boolean>(IframeEventType.InjectKeyExportBundle, {
       value: bundle,
       keyFormat,
       organizationId,
+      address,
     });
   }
 
@@ -432,6 +458,9 @@ export class IframeStamper {
    * The bundle should be encrypted to the iframe's initial public key
    * Encryption should be performed with HPKE (RFC 9180).
    * This is used during the wallet export flow.
+   * @param {string} bundle - The encrypted wallet export bundle to inject
+   * @param {string} organizationId - The organization ID
+   * @returns {Promise<boolean>} Returns true on successful injection
    */
   async injectWalletExportBundle(
     bundle: string,
@@ -449,6 +478,10 @@ export class IframeStamper {
   /**
    * Function to inject an import bundle into the iframe
    * This is used to initiate either the wallet import flow or the private key import flow.
+   * @param {string} bundle - The import bundle to inject
+   * @param {string} organizationId - The organization ID
+   * @param {string} userId - The user ID
+   * @returns {Promise<boolean>} Returns true on successful injection
    */
   async injectImportBundle(
     bundle: string,
@@ -467,6 +500,7 @@ export class IframeStamper {
    * The bundle should be encrypted to Turnkey's Signer enclave's initial public key
    * Encryption should be performed with HPKE (RFC 9180).
    * This is used during the wallet import flow.
+   * @returns {Promise<string>} The encrypted wallet bundle
    */
   async extractWalletEncryptedBundle(): Promise<string> {
     return this.createRequest<string>(
@@ -480,6 +514,8 @@ export class IframeStamper {
    * Encryption should be performed with HPKE (RFC 9180).
    * The key format to encode the private key in before it's encrypted and imported: HEXADECIMAL or SOLANA. Defaults to HEXADECIMAL.
    * This is used during the private key import flow.
+   * @param {KeyFormat} [keyFormat] - The key format (HEXADECIMAL or SOLANA). Defaults to HEXADECIMAL
+   * @returns {Promise<string>} The encrypted key bundle
    */
   async extractKeyEncryptedBundle(keyFormat?: KeyFormat): Promise<string> {
     return this.createRequest<string>(
@@ -491,6 +527,8 @@ export class IframeStamper {
   /**
    * Function to apply settings on allowed parameters in the iframe
    * This is used to style the HTML element used for plaintext in wallet and private key import.
+   * @param {TIframeSettings} settings - The settings object containing styles to apply
+   * @returns {Promise<boolean>} Returns true on successful application
    */
   async applySettings(settings: TIframeSettings): Promise<boolean> {
     return this.createRequest<boolean>(IframeEventType.ApplySettings, {
@@ -500,6 +538,9 @@ export class IframeStamper {
 
   /**
    * Function to sign a payload with the underlying iframe
+   * @param {string} payload - The payload to sign
+   * @returns {Promise<TStamp>} Object containing stamp header name and value
+   * @throws {Error} When iframe public key is null (init() not called/awaited)
    */
   async stamp(payload: string): Promise<TStamp> {
     if (this.iframePublicKey === null) {
@@ -514,28 +555,42 @@ export class IframeStamper {
   }
 
   /**
-   * Function to sign a transaction using an embedded private key in-memory within an iframe
+   * Function to sign a message using an embedded private key in-memory within an iframe
    * Returns the signed message string
+   * @param {TSignableMessage} message - The message to sign with type (Ethereum or Solana)
+   * @param {string} address - [Optional] Address to sign with
+   * @returns {Promise<string>} The signed message string
    */
-  async signMessage(message: TSignableMessage): Promise<string> {
+  async signMessage(
+    message: TSignableMessage,
+    address?: string,
+  ): Promise<string> {
     return this.createRequest<string>(IframeEventType.SignMessage, {
       value: JSON.stringify(message),
+      address,
     });
   }
 
   /**
-   * Function to sign a message using an embedded private key in-memory within an iframe
+   * Function to sign a transaction using an embedded private key in-memory within an iframe
    * Returns the signed, serialized transaction payload
+   * @param {TSignableTransaction} transaction - The transaction to sign with type (Ethereum or Solana)
+   * @param {string} address - [Optional] Address to sign with
+   * @returns {Promise<string>} The signed, serialized transaction payload
    */
-  async signTransaction(transaction: TSignableTransaction): Promise<string> {
+  async signTransaction(
+    transaction: TSignableTransaction,
+    address?: string,
+  ): Promise<string> {
     return this.createRequest<string>(IframeEventType.SignTransaction, {
       value: JSON.stringify(transaction),
+      address,
     });
   }
 
   /**
    * Function to clear the iframe's in-memory embedded private key. For now, we assume that there will be only one private key at most.
-   * Returns boolean
+   * @returns {Promise<boolean>} Returns true on successful clearing
    */
   async clearEmbeddedPrivateKey(): Promise<boolean> {
     return this.createRequest<boolean>(IframeEventType.clearEmbeddedPrivateKey);