diff --git a/index.htm b/index.htm
index 663c393..7b3b26b 100644
--- a/index.htm
+++ b/index.htm
@@ -23,7 +23,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, \'row\')\nplt.axis([0, 5, 0, 19])\nplt.plot(np.unique(x), np.poly1d(np.polyfit(x, y, 1))\nplt.show()\n'" />
+    <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'" />
   </p>
 
 </div>
diff --git a/styles.css b/styles.css
index a27f8ce..3f5baea 100644
--- a/styles.css
+++ b/styles.css
@@ -56,3 +56,8 @@ td {
 .btn-module {
   background: #FFE873;
 }
+
+.btn-success {
+  background: #5cb85c;
+  border-color: green;
+}