From 292ef5464d18a6420abc5a39ac9964cb15f95c9f Mon Sep 17 00:00:00 2001
From: Luke Murphy <lukewm@riseup.net>
Date: Wed, 5 Aug 2020 08:23:38 +0200
Subject: [PATCH] Fix up README example

---
 README.md | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 95d6a90..db89840 100644
--- a/README.md
+++ b/README.md
@@ -20,19 +20,34 @@ $ pip install merkle-tree-stream
 ```python
 import hashlib
 
+from merkle_tree_stream import MerkleTreeGenerator
+
+
 def _leaf(node, roots=None):
     return hashlib.sha256(node.data).digest()
 
+
 def _parent(first, second):
     sha256 = hashlib.sha256()
     sha256.update(first.data)
     sha256.update(second.data)
     return sha256.digest()
 
-merkle = MerkleTreeGenerator(leaf=leaf, parent=parent)
+
+merkle = MerkleTreeGenerator(leaf=_leaf, parent=_parent)
 
 merkle.write(b"a")
 merkle.write(b"b")
 
-assert len(merkle) == 2 + 1
+print(merkle._nodes)
+```
+
+Output:
+
+```sh
+[
+ MerkleTreeNode(index=0, parent=1, size=1, data=b'a', hash=b'...'),
+ MerkleTreeNode(index=2, parent=1, size=1, data=b'b', hash=b'...'),
+ MerkleTreeNode(index=1, parent=3, size=2, data=b'', hash=b'...')
+]
 ```