diff --git a/.gitignore b/.gitignore
index 150e5c1..a66fc2a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,6 +2,7 @@ build/*
 dist/*
 env/*
 .vscode/
+.idea/
 .ipynb_checkpoints/
 config.json
 __pycache__/
diff --git a/inspect4py/cli.py b/inspect4py/cli.py
index 166b275..65a93d9 100644
--- a/inspect4py/cli.py
+++ b/inspect4py/cli.py
@@ -51,7 +51,7 @@ def __init__(self, path, out_control_flow_path, out_json_path, control_flow, abs
             self.bodyInfo = self.inspect_body()
             if control_flow:
                 self.out_control_flow_path = out_control_flow_path
-                self.controlFlowInfo = self.inspect_controlflow()
+                self.controlFlowInfo = self.inspect_controlflow("png")
             else:
                 self.controlFlowInfo = {}
             self.fileJson = self.file_json()
@@ -1290,7 +1290,7 @@ def main(input_path, output_dir, ignore_dir_pattern, ignore_file_pattern, requir
                             else:
                                 dir_info[out_dir].append(code_info.fileJson[0])
                     except:
-                        print("Error when processing " + f + ": ", sys.exc_info()[0])
+                        print("Error when processing " + f + ": ", sys.exc_info())
                         continue
 
         # Generate the call list of the Dir