diff --git a/client/html/index.html b/client/html/index.html
index f0533c8..806c75a 100644
--- a/client/html/index.html
+++ b/client/html/index.html
@@ -19,8 +19,8 @@
                     <div class="panel panel-default">
                         <div class="panel-heading">
                             <big>Visualizer</big>
-                            <div class="btn-group pull-right">
-                                <button class="btn btn-xs btn-default" id="forget">Forget</button>
+                            <div class="pull-right">
+                                <button class="btn btn-xs btn-danger" id="forget">Forget</button>
                                 <button class="btn btn-xs btn-default" id="learn">Learn</button>
                             </div>
                         </div>
diff --git a/client/scripts/application.js b/client/scripts/application.js
index e8b5713..2de4ceb 100644
--- a/client/scripts/application.js
+++ b/client/scripts/application.js
@@ -177,12 +177,13 @@
     }
 
     function setCustomized(customized) {
-        $('#forget').prop('disabled', customized);
         if (customized) {
             $('#customized').show();
+            $('#forget').hide();
         }
         else {
             $('#customized').hide();
+            $('#forget').show();
         }
     }