diff --git a/band-gap-prediction/band_gap_prediction.bkr b/band-gap-prediction/band_gap_prediction.bkr
index 48c0b724769542bef80788513d2fc91ac545d4a5..701ef16c45a9853d95b997387a846e92fe51b61d 100644
--- a/band-gap-prediction/band_gap_prediction.bkr
+++ b/band-gap-prediction/band_gap_prediction.bkr
@@ -206,7 +206,7 @@
                 },
                 "selectedType": "BeakerDisplay",
                 "elapsedTime": 0,
-                "height": 485
+                "height": 366
             },
             "evaluatorReader": true,
             "lineCount": 25,
@@ -234,7 +234,7 @@
                 },
                 "selectedType": "BeakerDisplay",
                 "elapsedTime": 0,
-                "height": 256
+                "height": 232
             },
             "evaluatorReader": true,
             "lineCount": 5,
@@ -291,7 +291,7 @@
                 },
                 "selectedType": "BeakerDisplay",
                 "elapsedTime": 0,
-                "height": 933
+                "height": 698
             },
             "evaluatorReader": true,
             "lineCount": 34,
@@ -528,7 +528,7 @@
                 },
                 "selectedType": "BeakerDisplay",
                 "elapsedTime": 0,
-                "height": 677
+                "height": 511
             },
             "evaluatorReader": true,
             "lineCount": 19,
@@ -564,7 +564,7 @@
                 },
                 "selectedType": "BeakerDisplay",
                 "elapsedTime": 0,
-                "height": 535
+                "height": 440
             },
             "evaluatorReader": true,
             "lineCount": 13,
@@ -830,7 +830,7 @@
                 "selectedType": "BeakerDisplay",
                 "elapsedTime": 0,
                 "state": {},
-                "height": 621
+                "height": 593
             },
             "evaluatorReader": true,
             "lineCount": 243,
@@ -852,6 +852,10 @@
                     "f = open(os.devnull, 'w')",
                     "sys.stderr = f",
                     "",
+                    "#Ignore deprecation warnings",
+                    "import warnings",
+                    "warnings.filterwarnings(\"ignore\", category=DeprecationWarning) ",
+                    "",
                     "from xgboost import XGBClassifier",
                     "",
                     "from sklearn.model_selection import train_test_split",
@@ -1130,8 +1134,7 @@
                     ")",
                     "fig = go.Figure(data=data, layout=layout)",
                     "iplot(fig)"
-                ],
-                "hidden": true
+                ]
             },
             "output": {
                 "state": {},
@@ -1160,10 +1163,10 @@
                 "pluginName": "Python3",
                 "shellId": "BB8F4984E7B84EF698BF5DFC74D72A9C",
                 "elapsedTime": 16630,
-                "height": 1326
+                "height": 1318
             },
             "evaluatorReader": true,
-            "lineCount": 288,
+            "lineCount": 292,
             "isError": false,
             "tags": "train"
         },
@@ -1234,7 +1237,7 @@
                 },
                 "selectedType": "BeakerDisplay",
                 "elapsedTime": 0,
-                "height": 544
+                "height": 379
             },
             "evaluatorReader": true,
             "lineCount": 19,