diff --git a/CHANGE_LOG.txt b/CHANGE_LOG.txt
index f7f6666..6f127ea 100644
--- a/CHANGE_LOG.txt
+++ b/CHANGE_LOG.txt
@@ -103,4 +103,8 @@ Fixing issues when getting displayHTML on some environments
 Version 0.0.19
 --------------
 Adding functions.daydiff as a direct replacement for datediff
-Fixing some test_cases
\ No newline at end of file
+Fixing some test_cases
+
+Version 0.0.20
+--------------
+Changing the implementation for notebook integration for dataframe to a more standard approach
diff --git a/README.md b/README.md
index 4b5660e..37434f8 100644
--- a/README.md
+++ b/README.md
@@ -405,11 +405,17 @@ if tables.count() > 5:
 ```
 If you dont specify a name you can still access the last result using `__df`.
 
-The standard IPython display does not render snowpark `dataframe`. You can use `df.show()` but by default something like: `display(df)` will just print: `<snowflake.snowpark.dataframe.DataFrame at 0x7fa67d099c40>`
+> NOTE: By default only 50 rows are displays. You can customize this limit for example to 100 rows with:
+```
+import snowpark_extensions
+snowpark_extensions.rows_limit = 100
+```
+
+You can configure Jupyter to run some imports and initialization code at the start of a notebook by creating a file called `startup.ipy` in the `~/.ipython/profile_default/startup` directory. 
 
-You can configure Jupyter to run some imports and initialization code at the start of a notebook by creating a file called `startup.ipy` in the `~/.ipython/profile_default/startup` directory. Any code written in this file will be executed when you start a new Jupyter notebook.
+Any code written in this file will be executed when you start a new Jupyter notebook.
 
-An [example startup.ipy](./startup.ipy) is provided
+An [example startup.ipy](https://github.com/MobilizeNet/snowpark-extensions-py/blob/main/startup.ipy) is provided
 
 
 ## Running notebooks in Snowpark
diff --git a/example_notebook_with_snowpark.ipynb b/example_notebook_with_snowpark.ipynb
index 28e3723..87281cf 100644
--- a/example_notebook_with_snowpark.ipynb
+++ b/example_notebook_with_snowpark.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -40,23 +40,24 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Failed to execute query [queryID: 01a9e504-0405-9751-0000-1c55029dae1a] select asdfad\n",
-      "000904 (42000): SQL compilation error: error line 1 at position 7\n",
+      "Failed to execute query [queryID: 01a9f02a-0405-90fd-0000-1c5502a00ac6]  SELECT count(1) AS \"COUNT(LITERAL())\" FROM (select asdfad) LIMIT 1\n",
+      "000904 (42000): SQL compilation error: error line 1 at position 51\n",
       "invalid identifier 'ASDFAD'\n"
      ]
     },
     {
      "data": {
       "text/html": [
-       "<div style=\"background-color: #f44336; color: white; padding: 16px;\"><strong>Error:</strong> <span id=\"error-message\">000904 (42000): SQL compilation error: error line 1 at position 7\n",
+       "<div style=\"background-color: #f44336; color: white; padding: 16px;\"><strong>Error:</strong> <span id=\"error-message\">000904 (42000): SQL compilation error: error line 1 at position 51\n",
        "invalid identifier 'ASDFAD'</span></div>"
       ],
       "text/plain": [
-       "<IPython.core.display.HTML object>"
+       "<snowflake.snowpark.dataframe.DataFrame at 0x7fadfc77a700>"
       ]
      },
+     "execution_count": 4,
      "metadata": {},
-     "output_type": "display_data"
+     "output_type": "execute_result"
     }
    ],
    "source": [
@@ -66,7 +67,28 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%%sql tables\n",
+    "select 'a1' A, 'b1' B union\n",
+    "select 'a2'  , 'b2'"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "if tables.count() > 100:\n",
+    "    print(\"There are many tables here\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
@@ -95,7 +117,7 @@
        "</table>"
       ],
       "text/plain": [
-       "<IPython.core.display.HTML object>"
+       "<snowflake.snowpark.dataframe.DataFrame at 0x7fadfc77afa0>"
       ]
      },
      "metadata": {},
@@ -103,33 +125,14 @@
     }
    ],
    "source": [
-    "%%sql tables\n",
-    "select 'a1' A, 'b1' B union\n",
-    "select 'a2'  , 'b2'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "if tables.count() > 100:\n",
-    "    print(\"There are many tables here\")"
+    "display(tables)"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Output restricted to 10 rows\n"
-     ]
-    },
     {
      "data": {
       "text/html": [
@@ -137,286 +140,26 @@
        "  <thead>\n",
        "    <tr style=\"text-align: right;\">\n",
        "      <th></th>\n",
-       "      <th>TABLE_CATALOG</th>\n",
-       "      <th>TABLE_SCHEMA</th>\n",
-       "      <th>TABLE_NAME</th>\n",
-       "      <th>TABLE_OWNER</th>\n",
-       "      <th>TABLE_TYPE</th>\n",
-       "      <th>IS_TRANSIENT</th>\n",
-       "      <th>CLUSTERING_KEY</th>\n",
-       "      <th>ROW_COUNT</th>\n",
-       "      <th>BYTES</th>\n",
-       "      <th>RETENTION_TIME</th>\n",
-       "      <th>SELF_REFERENCING_COLUMN_NAME</th>\n",
-       "      <th>REFERENCE_GENERATION</th>\n",
-       "      <th>USER_DEFINED_TYPE_CATALOG</th>\n",
-       "      <th>USER_DEFINED_TYPE_SCHEMA</th>\n",
-       "      <th>USER_DEFINED_TYPE_NAME</th>\n",
-       "      <th>IS_INSERTABLE_INTO</th>\n",
-       "      <th>IS_TYPED</th>\n",
-       "      <th>COMMIT_ACTION</th>\n",
-       "      <th>CREATED</th>\n",
-       "      <th>LAST_ALTERED</th>\n",
-       "      <th>AUTO_CLUSTERING_ON</th>\n",
-       "      <th>COMMENT</th>\n",
+       "      <th>A</th>\n",
+       "      <th>B</th>\n",
        "    </tr>\n",
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
        "      <th>0</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>AUDIT_STAGE</td>\n",
-       "      <td>T_AUDIT_ACCESS_DATASET_FACT</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:09:39.907000-07:00</td>\n",
-       "      <td>2022-09-29 07:09:40.017000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
+       "      <td>a1</td>\n",
+       "      <td>b1</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>1</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>AUDIT_STAGE</td>\n",
-       "      <td>T_AUDIT_ACCESS_QUERY_FACT</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:09:40.420000-07:00</td>\n",
-       "      <td>2022-09-29 07:09:40.542000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>2</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>AUDIT_STAGE</td>\n",
-       "      <td>T_AUDIT_QUERY_DATASET_GROUP</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:09:40.469000-07:00</td>\n",
-       "      <td>2022-09-29 07:09:40.587000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>3</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>CJC</td>\n",
-       "      <td>V_WORKFORCE_INIT_ALL_JOB_RUNS</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>VIEW</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>NaN</td>\n",
-       "      <td>NaN</td>\n",
-       "      <td>NaN</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:54.934000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:55.665000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>4</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DBMS_SQL</td>\n",
-       "      <td>MYEMPLOYEES</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2021-11-24 21:02:07.868000-08:00</td>\n",
-       "      <td>2022-10-05 10:10:26.911000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>5</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>BIAUDIT_STAGE</td>\n",
-       "      <td>T_BI4_ADS_SERVER_TYPE_STR</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:09:32.081000-07:00</td>\n",
-       "      <td>2022-09-29 07:09:32.441000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>Source server type str data</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>6</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DIM</td>\n",
-       "      <td>L_BLOOD_URGENCY</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:32.102000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:33.220000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>Signifies the urgency associated with the blood test order</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>7</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DIM</td>\n",
-       "      <td>L_DNURSE_EPISODE_OUTCOME</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:19.989000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:21.669000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>Dimension table for holding reference information for the reasons for which the patient was discharged from the episode of care.</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>8</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DIM</td>\n",
-       "      <td>L_BLOOD_RETURN_REASON</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:40.229000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:41.441000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>Reason for returning a previously issued blood donation product eg. Recalled by QA</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>9</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DIM</td>\n",
-       "      <td>L_MCN_RESULT_ACTION</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:29.768000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:30.883000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>Any actions detailed based on the test results. Developed originally for MCN datamart.</td>\n",
+       "      <td>a2</td>\n",
+       "      <td>b2</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>"
       ],
       "text/plain": [
-       "<IPython.core.display.HTML object>"
+       "<snowflake.snowpark.dataframe.DataFrame at 0x7fadfc77afa0>"
       ]
      },
      "metadata": {},
@@ -429,24 +172,43 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 11,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "50"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "import snowpark_extensions\n",
+    "display(snowpark_extensions.rows_limit)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "# an 'extended' display\n",
-    "from snowpark_extensions import display"
+    "snowpark_extensions.rows_limit = 5"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Output restricted to 10 rows\n"
+      "There are 1361 rows. Showing only 5 \n"
      ]
     },
     {
@@ -509,56 +271,6 @@
        "    <tr>\n",
        "      <th>1</th>\n",
        "      <td>DEMODB</td>\n",
-       "      <td>AUDIT_STAGE</td>\n",
-       "      <td>T_AUDIT_ACCESS_QUERY_FACT</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:09:40.420000-07:00</td>\n",
-       "      <td>2022-09-29 07:09:40.542000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>2</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>AUDIT_STAGE</td>\n",
-       "      <td>T_AUDIT_QUERY_DATASET_GROUP</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:09:40.469000-07:00</td>\n",
-       "      <td>2022-09-29 07:09:40.587000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>3</th>\n",
-       "      <td>DEMODB</td>\n",
        "      <td>CJC</td>\n",
        "      <td>V_WORKFORCE_INIT_ALL_JOB_RUNS</td>\n",
        "      <td>ACCOUNTADMIN</td>\n",
@@ -582,32 +294,7 @@
        "      <td>None</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>4</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DBMS_SQL</td>\n",
-       "      <td>MYEMPLOYEES</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2021-11-24 21:02:07.868000-08:00</td>\n",
-       "      <td>2022-10-05 10:10:26.911000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>5</th>\n",
+       "      <th>2</th>\n",
        "      <td>DEMODB</td>\n",
        "      <td>BIAUDIT_STAGE</td>\n",
        "      <td>T_BI4_ADS_SERVER_TYPE_STR</td>\n",
@@ -632,35 +319,10 @@
        "      <td>Source server type str data</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>6</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DIM</td>\n",
-       "      <td>L_BLOOD_URGENCY</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:32.102000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:33.220000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>Signifies the urgency associated with the blood test order</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>7</th>\n",
+       "      <th>3</th>\n",
        "      <td>DEMODB</td>\n",
        "      <td>DIM</td>\n",
-       "      <td>L_DNURSE_EPISODE_OUTCOME</td>\n",
+       "      <td>L_MCN_SCORE</td>\n",
        "      <td>ACCOUNTADMIN</td>\n",
        "      <td>BASE TABLE</td>\n",
        "      <td>NO</td>\n",
@@ -676,13 +338,13 @@
        "      <td>YES</td>\n",
        "      <td>YES</td>\n",
        "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:19.989000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:21.669000-07:00</td>\n",
+       "      <td>2022-09-29 07:09:56.537000-07:00</td>\n",
+       "      <td>2022-09-29 07:09:57.953000-07:00</td>\n",
        "      <td>NO</td>\n",
-       "      <td>Dimension table for holding reference information for the reasons for which the patient was discharged from the episode of care.</td>\n",
+       "      <td>Holds MCN score information sourced from NMCN CLEFT system</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>8</th>\n",
+       "      <th>4</th>\n",
        "      <td>DEMODB</td>\n",
        "      <td>DIM</td>\n",
        "      <td>L_BLOOD_RETURN_REASON</td>\n",
@@ -706,44 +368,21 @@
        "      <td>NO</td>\n",
        "      <td>Reason for returning a previously issued blood donation product eg. Recalled by QA</td>\n",
        "    </tr>\n",
-       "    <tr>\n",
-       "      <th>9</th>\n",
-       "      <td>DEMODB</td>\n",
-       "      <td>DIM</td>\n",
-       "      <td>L_MCN_RESULT_ACTION</td>\n",
-       "      <td>ACCOUNTADMIN</td>\n",
-       "      <td>BASE TABLE</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>None</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>None</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>YES</td>\n",
-       "      <td>None</td>\n",
-       "      <td>2022-09-29 07:10:29.768000-07:00</td>\n",
-       "      <td>2022-09-29 07:10:30.883000-07:00</td>\n",
-       "      <td>NO</td>\n",
-       "      <td>Any actions detailed based on the test results. Developed originally for MCN datamart.</td>\n",
-       "    </tr>\n",
        "  </tbody>\n",
        "</table>"
       ],
       "text/plain": [
-       "<IPython.core.display.HTML object>"
+       "<snowflake.snowpark.dataframe.DataFrame at 0x7fa25b41e9d0>"
       ]
      },
+     "execution_count": 13,
      "metadata": {},
-     "output_type": "display_data"
+     "output_type": "execute_result"
     }
    ],
    "source": [
-    "display(tables)"
+    "%%sql\n",
+    "select * from information_schema.tables"
    ]
   }
  ],
diff --git a/runner b/runner
index 6b39793..cb76244 100755
--- a/runner
+++ b/runner
@@ -26,7 +26,7 @@ session.sql(f"CREATE STAGE IF NOT EXISTS {args.stage}").show()
 session.file.put(args.notebook,f'@{args.stage}',auto_compress=False,overwrite=True)
 
 
-packages=["snowflake-snowpark-python","nbconvert","nbformat","ipython","jinja2==3.0.3","plotly"] #"stack_data","pexpect","pickleshare","backcall","decorator","prompt_toolkit","jedi"
+packages=["snowflake-snowpark-python","nbconvert","nbformat","ipython","jinja2==3.0.3","plotly"]
 packages.extend(set(filter(None, args.packages.split(','))))
 print(f"Using packages [magenta]{packages}")
 
diff --git a/setup.py b/setup.py
index 2fbd263..2c4c178 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()
 
-VERSION = '0.0.19'
+VERSION = '0.0.20'
 
 setup(name='snowpark_extensions',
       version=VERSION,
diff --git a/snowpark_extensions/__init__.py b/snowpark_extensions/__init__.py
index 2c6cee3..cd216b0 100644
--- a/snowpark_extensions/__init__.py
+++ b/snowpark_extensions/__init__.py
@@ -6,9 +6,10 @@
 from .session_builder_extensions import *
 from .types_extensions import *
 from .column_extensions import *
-from .utils import display
 
 
+rows_limit = 50
+
 def get_display_html() -> None:
     import inspect
     for frame in inspect.getouterframes(inspect.currentframe()):
@@ -36,6 +37,29 @@ def instructions():
     except:
         from IPython.display import display, HTML
         output_cell_output = lambda x: display(HTML(x))
+
+    def snowflake_dataframe_formatter(df):
+        # Format the dataframe as a table using pandas
+        from snowflake.snowpark.exceptions import SnowparkSQLException
+        try:
+            count = df.count()
+            if count > rows_limit:
+                print(f"There are {count} rows. Showing only {rows_limit} ")
+            return df.limit(rows_limit).to_pandas().to_html()
+        except SnowparkSQLException as sce:
+            error_msg = sce.message
+            formatted = error_message_template.replace("@error", error_msg)
+            return formatted
+        except Exception as ex:
+            error_message = str(ex)
+            return f"<pre>{error_message}</pre>"
+   
+
+
+    # Register the display hook
+    from snowflake.snowpark import DataFrame
+    get_ipython().display_formatter.formatters['text/html'].for_type(DataFrame, snowflake_dataframe_formatter)
+
     from IPython.core.magic import (Magics, magics_class, cell_magic)
     @magics_class
     class SnowparkMagics(Magics):
@@ -51,22 +75,12 @@ def sql(self, line, cell):
                 name = None
                 if line and line.strip():
                     name = line.strip().split(" ")[0]
-                from snowflake.snowpark.exceptions import SnowparkSQLException
-                try:
-                    df = session.sql(res)
-                    html = df.to_pandas().to_html()
-                    output_cell_output(html)
-                    if name:
-                        self.shell.user_ns[name] = df
-                    else:
-                        self.shell.user_ns["__df"] = df
-                except SnowparkSQLException as sce:
-                    error_msg = sce.message
-                    formatted = error_message_template.replace("@error", error_msg)
-                    output_cell_output(formatted)
-                except Exception as ex:
-                    error_message = str(ex)
-                    output_cell_output(f"<pre>{error_message}</pre>")                    
+                df = session.sql(res)
+                if name:
+                    self.shell.user_ns[name] = df
+                else:
+                    self.shell.user_ns["__df"] = df
+                    return df
             else:
                 return "No session was found. You can setup one by running: session = Session.builder.from_env().getOrCreate()"
     magics = SnowparkMagics(ipython)
diff --git a/snowpark_extensions/utils.py b/snowpark_extensions/utils.py
index eafaa96..6a1a218 100644
--- a/snowpark_extensions/utils.py
+++ b/snowpark_extensions/utils.py
@@ -149,20 +149,3 @@ def schema_str_to_schema(schema_as_str):
         datatype = map_string_type_to_datatype(type)
         schema_fields.append(StructField(name,datatype))
     return StructType(schema_fields)
-
-from snowflake.snowpark import DataFrame
-def display(data:DataFrame, limit:int=10):
-    from IPython.display import display as _display, HTML
-    if isinstance(data, DataFrame):
-        try:
-            output_cell_output = displayHTML
-        except:
-            output_cell_output = lambda x: _display(HTML(x))
-        try:
-            print(f"Output restricted to {limit} rows")
-            html = data.limit(limit).to_pandas().to_html()
-            output_cell_output(html)
-        except Exception as e:
-            print(f"Error displaying dataframe {e}")
-    else:
-        _display(data)
\ No newline at end of file