2 files changed, 6 insertions, 4 deletions

diff --git a/src/Network/BitTorrent/DHT/Protocol.hs b/src/Network/BitTorrent/DHT/Protocol.hs
index 73e5fa68..0ac814b7 100644
--- a/src/Network/BitTorrent/DHT/Protocol.hs
+++ b/src/Network/BitTorrent/DHT/Protocol.hs
@@ -42,7 +42,7 @@ import Network.BitTorrent.Exchange.Protocol ()
 
 type NodeId = ByteString
 
--- WARN is the 'system' random suitable for this?
+-- TODO WARN is the 'system' random suitable for this?
 -- | Generate random NodeID used for the entire session.
 --   Distribution of ID's should be as uniform as possible.
 --
@@ -56,9 +56,7 @@ data NodeAddr = NodeAddr {
 
 instance Serialize NodeAddr where
   get = NodeAddr <$> getWord32be <*> get
-  put NodeAddr {..} = do
-    putWord32be nodeIP
-    put         nodePort
+  put NodeAddr {..} = putWord32be nodeIP >> put nodePort
 
 data NodeInfo = NodeInfo {
     nodeID   :: !NodeId
diff --git a/src/Network/BitTorrent/Discovery.hs b/src/Network/BitTorrent/Discovery.hs
new file mode 100644
index 00000000..f26864a7
--- /dev/null
+++ b/src/Network/BitTorrent/Discovery.hs
@@ -0,0 +1,4 @@
+type PeerCount = Int
+
+defaultChanSize :: PeerCount
+defaultChanSize = defaultNumWant * 2