diff --git a/lantern/extensions/hideinput/extension.py b/lantern/extensions/hideinput/extension.py
index d8bd53b..c3290f7 100644
--- a/lantern/extensions/hideinput/extension.py
+++ b/lantern/extensions/hideinput/extension.py
@@ -6,4 +6,3 @@ def load_jupyter_server_extension(nb_server_app):
     exporter_map['pdf_hidecode'] = PDFHideCodeExporter
     exporter_map['html_hidecode'] = HTMLHideCodeExporter
     print('Installing lantern nbconvert endpoints')
-    import pdb; pdb.set_trace()
diff --git a/lantern/plotting/plot_matplotlib.py b/lantern/plotting/plot_matplotlib.py
index 2474146..3d45592 100644
--- a/lantern/plotting/plot_matplotlib.py
+++ b/lantern/plotting/plot_matplotlib.py
@@ -236,6 +236,9 @@ def plot(data, **kwargs):
             _MFA[0].set_ylabel(kwargs.get('ylabel'))
         if 'title' in kwargs:
             plt.title(kwargs.get('title'))
+
+        if kwargs.get('raw', False):
+            return plt
         return plt.show()
 
     @staticmethod