SeannLin
diff --git a/‎README.rst‎
Lines changed: 15 additions & 1 deletion b/‎README.rst‎
Lines changed: 15 additions & 1 deletion
diff --git a/‎hide_code/hide_code.js‎
Lines changed: 21 additions & 6 deletions b/‎hide_code/hide_code.js‎
Lines changed: 21 additions & 6 deletions
diff --git a/‎setup.py‎
Lines changed: 1 addition & 1 deletion b/‎setup.py‎
Lines changed: 1 addition & 1 deletion
@@ -20,23 +20,37 @@ hide_code is an extension for Jupyter/IPython notebooks to selectively hide code
 Normal
 ------
 .. image:: https://raw.githubusercontent.com/kirbs-/hide_code/dev/images/1.1.png
-   :scale: 50 %
+   :width: 918
+   :height: 322
+   :scale: 70 
 
 Hide code enabled
 -----------------
 .. image:: https://raw.githubusercontent.com/kirbs-/hide_code/dev/images/1.2.png
+   :width: 1227
+   :height: 322
+   :scale: 60 
 
 Or customize each cell by selecting "Hide code" from the Cell Toolbar dropdown. Then use "Hide Code" and "Hide Prompts" checkboxes to hide the specific cell's code or cell's input/output prompts.
 
 .. image:: https://raw.githubusercontent.com/kirbs-/hide_code/dev/images/2.png
+   :width: 408
+   :height: 242
+   :scale: 60 
 
 Normal
 ------
 .. image:: https://raw.githubusercontent.com/kirbs-/hide_code/dev/images/3.1.png
+   :width: 696   
+   :height: 322
+   :scale: 90 
 
 Hide code enabled
 -----------------
 .. image:: https://raw.githubusercontent.com/kirbs-/hide_code/dev/images/3.2.png
+   :width: 764
+   :height: 322
+   :scale: 80 
 
 Installation
 ============
 
@@ -43,18 +43,18 @@ function ($, celltoolbar){
 
 	function toggleHideCode(cell){
 		var c = $(cell.element);
-		if (cell.metadata.hideCode && cell.classs_config.classname != 'MarkdownCell'){
+		if (cell.metadata.hideCode && cell.class_config.classname != 'MarkdownCell'){
 			c.find('.input_area').hide(); 
-		} else {
+		} else if(cell.class_config.classname != 'MarkdownCell') {
 			c.find('.input_area').show(); 
 		}
 	}
 
 	function toggleHidePrompt(cell){
 		var c = $(cell.element);
-		if (cell.metadata.hidePrompt){
+		if (cell.metadata.hidePrompt && cell.class_config.classname != 'MarkdownCell'){
 			c.find('.prompt').css('visibility','hidden');
-		} else {
+		} else if(cell.class_config.classname != 'MarkdownCell') {
 			c.find('.prompt').css('visibility','visible'); 
 		}
 	}
@@ -67,10 +67,25 @@ function ($, celltoolbar){
 		    {
 		     'label' : 'Hide/show code',
 		     'icon' : 'fa-code',
-		     'callback' : function() { 
+		     'callback' : function() { // toggling visibility is adding display: blcok to the element. Causing celltoolbar = None not work.
 		        $('.input').toggle(); 
 		        $('.prompt').toggle(); 
-		        $('.ctb_show').toggle();
+		        var ctb = $('.ctb_hideshow');
+		        if(ctb.hasClass('ctb_show')) {
+		        	ctb.removeClass('ctb_show');
+		        } else {
+		        	ctb.addClass('ctb_show');
+		        }
+
+		        var ctb = $('.celltoolbar');
+		        if(ctb.hasClass('invisible')){
+		        	console.log('visible');
+		        	ctb.removeClass('invisible');
+		        } else {
+		        	console.log('invisible');
+		        	ctb.addClass('invisible');
+		        }
+		        // $('.celltoolbar').toggle();
 		        $('.selected').removeClass('selected').addClass('unselected');
 		      } 
 		    }
 
@@ -31,7 +31,7 @@ def run(self):
     # Versions should comply with PEP440.  For a discussion on single-sourcing
     # the version across setup.py and the project code, see
     # https://packaging.python.org/en/latest/single_source_version.html
-    version='0.1.5',
+    version='0.1.6',
 
     description='A Jupyter notebook extension to hide code.',
     long_description=long_description,