diff --git a/main.ipynb b/main.ipynb
index f8d7d2d..afa552e 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -2,9 +2,11 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "e706cfb0-2234-4c4c-95d8-d1968f656aa0",
-   "metadata": {},
+   "metadata": {
+    "tags": []
+   },
    "outputs": [],
    "source": [
     "from datetime import datetime, timedelta\n",
@@ -77,6 +79,7 @@
     "    total_volume AS \"Total Volume\"\n",
     "FROM\n",
     "    time_windows\n",
+    "    limit 1000\n",
     "\"\"\"\n",
     "\n",
     "\n",
@@ -113,6 +116,17 @@
     "        # final_scenario_df['RUN_DATE'] = final_scenario_df['END_DATE']\n",
     "        return final_scenario_df\n"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b5c4307f-bc35-47e2-b680-fd1df2168d6c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "scenario = Scenario()\n",
+    "scenario.logic()"
+   ]
   }
  ],
  "metadata": {
diff --git a/main.py b/main.py
index 04fdca1..1efd05c 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # coding: utf-8
 
-# In[ ]:
+# In[2]:
 
 
 from datetime import datetime, timedelta
@@ -74,6 +74,7 @@ SELECT
     total_volume AS "Total Volume"
 FROM
     time_windows
+    limit 1000
 """
 
 
@@ -110,3 +111,10 @@ class Scenario:
         # final_scenario_df['RUN_DATE'] = final_scenario_df['END_DATE']
         return final_scenario_df
 
+
+# In[ ]:
+
+
+scenario = Scenario()
+scenario.logic()
+