diff --git a/src/jupyter_contrib_nbextensions/nbextensions/zenmode/main.js b/src/jupyter_contrib_nbextensions/nbextensions/zenmode/main.js
index 8a1cb43..270e9f4 100644
--- a/src/jupyter_contrib_nbextensions/nbextensions/zenmode/main.js
+++ b/src/jupyter_contrib_nbextensions/nbextensions/zenmode/main.js
@@ -156,7 +156,7 @@ define([
             'zenmode-btn-grp'
         );
         $("#maintoolbar-container").prepend($('#zenmode-btn-grp'));
-        return IPython.notebook.config.loaded.then(initialize);
+        config.load();
     };
 
     var extension = {
diff --git a/src/jupyter_contrib_nbextensions/nbextensions/zenmode/zenmode.yaml b/src/jupyter_contrib_nbextensions/nbextensions/zenmode/zenmode.yaml
index 8b1ccae..9f2a587 100644
--- a/src/jupyter_contrib_nbextensions/nbextensions/zenmode/zenmode.yaml
+++ b/src/jupyter_contrib_nbextensions/nbextensions/zenmode/zenmode.yaml
@@ -3,7 +3,7 @@ Name: zenmode
 Link: README.md
 Description: A little extension to give Zenmode functionality to the IPython notebook
 Main: main.js
-Compatibility: 4.x
+Compatibility: 4.x, 5.x
 Parameters:
 - name: zenmode_set_zenmode_on_load
   description: Set zenmode on when a notebook opens
@@ -13,6 +13,14 @@ Parameters:
   description: Use builtin backgrounds in addition to any specified by URL
   input_type: checkbox
   default: true
+- name: zenmode_hide_header
+  description: Hide the header in zenmode
+  input_type: checkbox
+  default: true
+- name: zenmode_hide_menubar
+  description: Hide the menubar in zenmode
+  input_type: checkbox
+  default: true
 - name: zenmode_backgrounds
   description: "Urls to use as backgrounds. Any beginning with # are ignored."
   input_type: list