diff --git a/main.py b/main.py
index 436eaa1..5232101 100644
--- a/main.py
+++ b/main.py
@@ -27,11 +27,11 @@ async def dm():
             f"\nFrom {event.public_key[:3]}..{event.public_key[-3:]}: {decrypted_content}"
         )
 
-    client = NostrClient(privatekey_hex=pk)
+    client = NostrClient(private_key=pk)
     if not pk:
-        print(f"Your private key: {client.private_key.hex()}")
+        print(f"Your private key: {client.private_key.bech32()}")
 
-    print(f"Your public key: {client.public_key.hex()}")
+    print(f"Your public key: {client.public_key.bech32()}")
 
     # await asyncio.sleep(1)
     t = threading.Thread(
@@ -43,16 +43,20 @@ async def dm():
     )
     t.start()
 
-    to_pubk_hex = (
-        input("Enter other pubkey to post to (enter nothing to DM yourself): ")
+    pubkey_to_str = (
+        input("Enter other pubkey to DM to (enter nothing to DM yourself): ")
         or client.public_key.hex()
     )
-    print(f"Subscribing to DMs to {to_pubk_hex}")
+    if pubkey_to_str.startswith("npub"):
+        pubkey_to = PublicKey().from_npub(pubkey_to_str)
+    else:
+        pubkey_to = PublicKey(bytes.fromhex(pubkey_to_str))
+    print(f"Sending DMs to {pubkey_to.bech32()}")
     while True:
         print_status(client)
         await asyncio.sleep(1)
         msg = input("\nEnter message: ")
-        client.dm(msg, PublicKey(bytes.fromhex(to_pubk_hex)))
+        client.dm(msg, pubkey_to)
 
 
 async def post():
@@ -67,20 +71,23 @@ async def post():
             f"\nFrom {event.public_key[:3]}..{event.public_key[-3:]}: {event.content}"
         )
 
-    sender_client = NostrClient(privatekey_hex=pk)
+    sender_client = NostrClient(private_key=pk)
     # await asyncio.sleep(1)
 
-    to_pubk_hex = (
+    pubkey_to_str = (
         input(
             "Enter other pubkey (enter nothing to read your own posts, enter * for all): "
         )
         or sender_client.public_key.hex()
     )
-    if to_pubk_hex == "*":
-        to_pubk = None
+    if pubkey_to_str == "*":
+        pubkey_to = None
+    elif pubkey_to_str.startswith("npub"):
+        pubkey_to = PublicKey().from_npub(pubkey_to_str)
     else:
-        print(f"Subscribing to posts by {to_pubk_hex}")
-        to_pubk = PublicKey(bytes.fromhex(to_pubk_hex))
+        pubkey_to = PublicKey(bytes.fromhex(pubkey_to_str))
+
+    print(f"Subscribing to posts by {pubkey_to.bech32() if pubkey_to else 'everyone'}")
 
     filters = {
         "since": int(
@@ -93,7 +100,7 @@ async def post():
     t = threading.Thread(
         target=sender_client.get_post,
         args=(
-            to_pubk,
+            pubkey_to,
             callback,
             filters,
         ),
@@ -107,9 +114,9 @@ async def post():
         sender_client.post(msg)
 
 
-if input("Enter '1' for DM, '2' for Posts (Default: 1):") or 1 == 1:
-    # write a DM and receive DMs
-    asyncio.run(dm())
-else:
+if input("Enter '1' for DM, '2' for Posts (Default: 1): ") == "2":
     # make a post and subscribe to posts
     asyncio.run(post())
+else:
+    # write a DM and receive DMs
+    asyncio.run(dm())
diff --git a/nostr/client/client.py b/nostr/client/client.py
index 169ccc1..0c246d8 100644
--- a/nostr/client/client.py
+++ b/nostr/client/client.py
@@ -30,8 +30,8 @@ class NostrClient:
     private_key: PrivateKey
     public_key: PublicKey
 
-    def __init__(self, privatekey_hex: str = "", relays: List[str] = [], connect=True):
-        self.generate_keys(privatekey_hex)
+    def __init__(self, private_key: str = "", relays: List[str] = [], connect=True):
+        self.generate_keys(private_key)
 
         if len(relays):
             self.relays = relays
@@ -48,9 +48,13 @@ class NostrClient:
     def close(self):
         self.relay_manager.close_connections()
 
-    def generate_keys(self, privatekey_hex: str = None):
-        pk = bytes.fromhex(privatekey_hex) if privatekey_hex else None
-        self.private_key = PrivateKey(pk)
+    def generate_keys(self, private_key: str = None):
+        if private_key.startswith("nsec"):
+            self.private_key = PrivateKey.from_nsec(private_key)
+        elif private_key:
+            self.private_key = PrivateKey(bytes.fromhex(private_key))
+        else:
+            self.private_key = PrivateKey()  # generate random key
         self.public_key = self.private_key.public_key
 
     def post(self, message: str):