diff --git a/src/hdf5_vis.py b/src/hdf5_vis.py
index 036f9bc..b8aeddf 100644
--- a/src/hdf5_vis.py
+++ b/src/hdf5_vis.py
@@ -14,6 +14,7 @@ import pandas as pd
 from plotly.subplots import make_subplots
 import plotly.graph_objects as go
 import plotly.express as px
+#import plotly.io as pio
 
 def display_group_hierarchy_on_a_treemap(filename: str):
 
@@ -54,6 +55,8 @@ def display_group_hierarchy_on_a_treemap(filename: str):
     fig.show()
     file_name, file_ext = os.path.splitext(filename)
     fig.write_html(file_name + ".html")
+   
+    #pio.write_image(fig,file_name + ".png",width=800,height=600,format='png')
 
 #