diff --git a/src/js/app.js b/src/js/app.js
index 175ed11..b32ce2d 100644
--- a/src/js/app.js
+++ b/src/js/app.js
@@ -114,7 +114,11 @@ module.exports = new Vue({
         motors: [{}, {}, {}, {}],
         version: '<loading>'
       },
-      state: {messages: []},
+      state: {
+        messages: [],
+        wait_for_probing_complete: false,
+        show_probe_complete_modal: false
+      },
       video_size: cookie.get('video-size', 'small'),
       crosshair: cookie.get('crosshair', 'false') != 'false',
       errorTimeout: 30,
diff --git a/src/js/message.js b/src/js/message.js
index a720511..57c1fae 100644
--- a/src/js/message.js
+++ b/src/js/message.js
@@ -36,6 +36,12 @@ module.exports = {
       type: Boolean,
       required: true,
       twoWay: true
+    },
+
+    class: {
+      type: String,
+      required: false,
+      twoWay: false
     }
   }
 }
diff --git a/src/pug/templates/message.pug b/src/pug/templates/message.pug
index 2b724ad..7257547 100644
--- a/src/pug/templates/message.pug
+++ b/src/pug/templates/message.pug
@@ -26,7 +26,7 @@
 //-/////////////////////////////////////////////////////////////////////////////
 
 script#message-template(type="text/x-template")
-  .modal-mask(v-if="show")
+  .modal-mask(v-if="show", :class="class")
     .modal-wrapper
       .modal-container
         .modal-header