diff --git a/src/index.ts b/src/index.ts
index d3a1d81..1b3a6bd 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,10 +1,10 @@
 import {
-    callModule as callModuleKernel,
-    connectModule as connectModuleKernel,
+  callModule as callModuleKernel,
+  connectModule as connectModuleKernel,
 } from "libkernel";
 import {
-    callModule as callModuleModule,
-    connectModule as connectModuleModule,
+  callModule as callModuleModule,
+  connectModule as connectModuleModule,
 } from "libkmodule";
 import { EventEmitter } from "events";
 import { DataFn, ErrTuple } from "libskynet";
@@ -13,82 +13,89 @@ import { Buffer } from "buffer";
 const DHT_MODULE = "AQD1IgE4lTZkq1fqdoYGojKRNrSk0YQ_wrHbRtIiHDrnow";
 
 let callModule: typeof callModuleModule,
-    connectModule: typeof connectModuleModule;
+  connectModule: typeof connectModuleModule;
 
 if (window.document) {
-    callModule = callModuleKernel;
-    connectModule = connectModuleKernel;
+  callModule = callModuleKernel;
+  connectModule = connectModuleKernel;
 } else {
-    callModule = callModuleModule;
-    connectModule = connectModuleModule;
+  callModule = callModuleModule;
+  connectModule = connectModuleModule;
 }
 
 export class DHT {
-    public async connect(pubkey: string): Promise<Socket> {
-        const [resp, err] = await callModule(DHT_MODULE, "connect", { pubkey });
-        if (err) {
-            throw err;
-        }
-        return new Socket(resp.id);
+  public async connect(pubkey: string): Promise<Socket> {
+    const [resp, err] = await callModule(DHT_MODULE, "connect", { pubkey });
+    if (err) {
+      throw err;
     }
+    return new Socket(resp.id);
+  }
 
-    async ready(): Promise<ErrTuple> {
-        return callModule(DHT_MODULE, "ready");
+  async ready(): Promise<ErrTuple> {
+    return callModule(DHT_MODULE, "ready");
+  }
+
+  public async addRelay(pubkey: string): Promise<void> {
+    const [, err] = await callModule(DHT_MODULE, "addRelay", { pubkey });
+    if (err) {
+      throw err;
     }
+  }
 }
 
 export class Socket extends EventEmitter {
-    private id: number;
-    private eventUpdates: { [event: string]: DataFn[] } = {};
+  private id: number;
+  private eventUpdates: { [event: string]: DataFn[] } = {};
 
-    constructor(id: number) {
-        super();
-        this.id = id;
+  constructor(id: number) {
+    super();
+    this.id = id;
+  }
+
+  on(eventName: string, listener: (...args: any[]) => void): this {
+    const [update, promise] = connectModule(
+      DHT_MODULE,
+      "listenSocketEvent",
+      { id: this.id, event: eventName },
+      (data: any) => {
+        this.emit(eventName, data);
+      }
+    );
+    this.trackEvent(eventName, update);
+
+    promise.then(() => {
+      this.off(eventName, listener);
+    });
+
+    return super.on(eventName, listener);
+  }
+
+  off(type: string, listener: any): this {
+    const updates = [...this.eventUpdates[type]];
+    this.eventUpdates[type] = [];
+    for (const func of updates) {
+      func({ action: "off" });
     }
+    return super.off(type, listener);
+  }
 
-    on(eventName: string, listener: (...args: any[]) => void): this {
-        const [update, promise] = connectModule(
-            DHT_MODULE,
-            "listenSocketEvent",
-            { id: this.id, event: eventName },
-            (data: any) => {
-                this.emit(eventName, data);
-            }
-        );
-        this.trackEvent(eventName, update);
+  write(message: string | Buffer): void {
+    callModule(DHT_MODULE, "write", { id: this.id, message });
+  }
 
-        promise.then(() => {
-            this.off(eventName, listener);
-        });
+  end(): void {
+    callModule(DHT_MODULE, "close", { id: this.id });
+  }
 
-        return super.on(eventName, listener);
+  private ensureEvent(event: string): void {
+    if (!(event in this.eventUpdates)) {
+      this.eventUpdates[event] = [];
     }
+  }
 
-    off(type: string, listener: any): this {
-        const updates = [...this.eventUpdates[type]];
-        this.eventUpdates[type] = [];
-        for (const func of updates) {
-            func({ action: "off" });
-        }
-        return super.off(type, listener);
-    }
-
-    write(message: string | Buffer): void {
-        callModule(DHT_MODULE, "write", { id: this.id, message });
-    }
-
-    end(): void {
-        callModule(DHT_MODULE, "close", { id: this.id });
-    }
-
-    private ensureEvent(event: string): void {
-        if (!(event in this.eventUpdates)) {
-            this.eventUpdates[event] = [];
-        }
-    }
-
-    private trackEvent(event: string, update: DataFn): void {
-        this.ensureEvent(event as string);
-        this.eventUpdates[event].push(update);
-    }
+  private trackEvent(event: string, update: DataFn): void {
+    this.ensureEvent(event as string);
+    this.eventUpdates[event].push(update);
+  }
 }