From 13ef6c803e3ea8fdbad59007db6ae2c31dd161f9 Mon Sep 17 00:00:00 2001
From: Donne Martin <donne.martin@gmail.com>
Date: Thu, 9 Feb 2017 05:20:53 -0500
Subject: [PATCH] Update bst challenge control flow (#146)

---
 graphs_trees/bst/bst.py             | 3 ++-
 graphs_trees/bst/bst_solution.ipynb | 5 +++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/graphs_trees/bst/bst.py b/graphs_trees/bst/bst.py
index b95fe2f..9b23e8e 100644
--- a/graphs_trees/bst/bst.py
+++ b/graphs_trees/bst/bst.py
@@ -21,7 +21,8 @@ class Bst(object):
         if self.root is None:
             self.root = Node(data)
             return self.root
-        return self._insert(self.root, data)
+        else:
+            return self._insert(self.root, data)
 
     def _insert(self, node, data):
         if node is None:
diff --git a/graphs_trees/bst/bst_solution.ipynb b/graphs_trees/bst/bst_solution.ipynb
index 914acdd..fb3bc5a 100644
--- a/graphs_trees/bst/bst_solution.ipynb
+++ b/graphs_trees/bst/bst_solution.ipynb
@@ -136,7 +136,8 @@
     "        if self.root is None:\n",
     "            self.root = Node(data)\n",
     "            return self.root\n",
-    "        return self._insert(self.root, data)\n",
+    "        else:\n",
+    "            return self._insert(self.root, data)\n",
     "\n",
     "    def _insert(self, node, data):\n",
     "        if node is None:\n",
@@ -292,7 +293,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.0"
+   "version": "3.4.3"
   }
  },
  "nbformat": 4,