diff --git a/rent-or-buy.ipynb b/rent-or-buy.ipynb
index bbc585e..59eedb3 100644
--- a/rent-or-buy.ipynb
+++ b/rent-or-buy.ipynb
@@ -49,13 +49,17 @@
    ],
    "source": [
     "\n",
+      "import math\n",
+    "import random\n",
+    "import time\n",
+    "\n",
+    "from IPython.display import display",
     "from ipywidgets import *\n",
     "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
-    "import time\n",
+       "\n",
     "%matplotlib notebook\n",
-    "import math\n",
-    "import random\n",
+ 
     "\n",
     "time = np.linspace(2019, 2060,30)\n",
     "\n",