4 files changed, 4 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Client.hs b/src/Network/BitTorrent/Client.hs
index 0d2e14ca..9b53ad6d 100644
--- a/src/Network/BitTorrent/Client.hs
+++ b/src/Network/BitTorrent/Client.hs
@@ -112,6 +112,7 @@ closeClient :: Client -> IO ()
 closeClient Client {..} = do
   Exchange.closeManager exchangeManager
   Tracker.closeManager  trackerManager
+  DHT.stopNode clientNode
   return ()
 --  closeNode clientNode
 
diff --git a/src/Network/BitTorrent/DHT.hs b/src/Network/BitTorrent/DHT.hs
index 6c78d992..080a0fc0 100644
--- a/src/Network/BitTorrent/DHT.hs
+++ b/src/Network/BitTorrent/DHT.hs
@@ -45,6 +45,7 @@ module Network.BitTorrent.DHT
        , Node
        , handlers
        , startNode
+       , stopNode
 
          -- ** Monad
        , MonadDHT (..)
diff --git a/src/Network/BitTorrent/DHT/Query.hs b/src/Network/BitTorrent/DHT/Query.hs
index 07fb80c2..714e9559 100644
--- a/src/Network/BitTorrent/DHT/Query.hs
+++ b/src/Network/BitTorrent/DHT/Query.hs
@@ -80,6 +80,7 @@ announceH = nodeHandler $ \ naddr @ NodeAddr {..} (Announce {..}) -> do
   insertPeer topic peerAddr
   return Announced
 
+-- | Includes all query handlers.
 handlers :: Address ip => [NodeHandler ip]
 handlers = [pingH, findNodeH, getPeersH, announceH]
 
diff --git a/src/Network/BitTorrent/DHT/Session.hs b/src/Network/BitTorrent/DHT/Session.hs
index 1bc9e697..b4ff0208 100644
--- a/src/Network/BitTorrent/DHT/Session.hs
+++ b/src/Network/BitTorrent/DHT/Session.hs
@@ -308,6 +308,7 @@ startNode hs opts naddr logger = do
 stopNode :: Node ip -> IO ()
 stopNode Node {..} = closeInternalState resources
 
+-- | Run DHT operation on the given session.
 runDHT :: Node ip -> DHT ip a -> IO a
 runDHT node action = runReaderT (unDHT action) node
 {-# INLINE runDHT #-}