diff --git a/index.htm b/index.htm
index 7b3b26b..bd5e033 100644
--- a/index.htm
+++ b/index.htm
@@ -24,6 +24,7 @@ <h2>Linear regression</h2>
   <p>
     <input type="button" class="btn btn-module" value="modules" onclick="document.editor.textbox.value+='\nfrom __future__ import absolute_import, division, print_function, unicode_literals\nimport matplotlib.pyplot as plt\nimport numpy as np\nimport pandas as pd\nfrom IPython.display import clear_output\nfrom six.moves import urllib\nimport tensorflow.compat.v2.feature_column as fc\nimport tensorflow as tf'" />
     <input type="button" class="btn btn-success" value="sample dataset" onclick="document.editor.textbox.value+='\nx = [1, 2, 3, 4, 5]\ny = [1, 3, 8, 10, 14]\nplt.plot(x, y, \'ro\')\nplt.axis([0, 6, 0, 15])\nx_unique = np.unique(x)\npoly_func = np.poly1d(np.polyfit(x, y, 1))\nplt.plot(x_unique, poly_func(x_unique))\nplt.show()\n'" />
+    <input type="button" class="btn btn-success" value="load dataset" onclick="document.editor.textbox.value+='\ndftrain = pd.read_csv(\'https://storage.googleapis.com/tf-datasets/titanic/train.csv\') # training data\ndfeval = pd.read_csv(\'https://storage.googleapis.com/tf-datasets/titanic/eval.csv\') # testing data\ny_train = dftrain.pop(\'survived\')\ny_eval = dfeval.pop(\'survived\')'" />
   </p>
 
 </div>
diff --git a/styles.css b/styles.css
index 3f5baea..ab5f12e 100644
--- a/styles.css
+++ b/styles.css
@@ -61,3 +61,7 @@ td {
   background: #5cb85c;
   border-color: green;
 }
+
+.btn-success {
+  color: white;
+}