diff --git a/www/whiteboard/main.js b/www/whiteboard/main.js
index 183295c1e..239a4f21f 100644
--- a/www/whiteboard/main.js
+++ b/www/whiteboard/main.js
@@ -50,6 +50,7 @@ define([
 
         $width.on('change', updateBrushWidth);
 
+        // TODO add a better color palette
         var palette = ['red', 'blue', 'green', 'white', 'black', 'purple',
             'gray', 'beige', 'brown', 'cyan', 'darkcyan', 'gold', 'yellow', 'pink'];
         var $colors = $('#colors');
@@ -175,6 +176,13 @@ define([
             var $export = Cryptpad.createButton('export', true, {}, saveImage);
             $rightside.append($export);
 
+            var $forget = Cryptpad.createButton('forget', true, {}, function (err, title) {
+                if (err) { return; }
+                setEditable(false);
+                toolbar.failed();
+            });
+            $rightside.append($forget);
+
             var editHash;
             var viewHash = Cryptpad.getViewHashFromKeys(info.channel, secret.keys);
 
@@ -339,13 +347,12 @@ define([
 
         var onAbort = config.onAbort = function (info) {
             setEditable(false);
-            window.alert("Server Connection Lost");
-
-            if (window.confirm("Would you like to save your image?")) {
-                saveImage();
-            }
+            toolbar.failed();
+            Cryptpad.alert(Messages.common_connectionLost, undefined, true);
         };
 
+        // TODO onConnectionStateChange
+
         var rt = Realtime.start(config);
 
         canvas.on('mouse:up', onLocal);