diff --git a/__pycache__/code_to_image.cpython-39.pyc b/__pycache__/code_to_image.cpython-39.pyc
index 41d516a..ac1d18e 100644
Binary files a/__pycache__/code_to_image.cpython-39.pyc and b/__pycache__/code_to_image.cpython-39.pyc differ
diff --git a/code_to_image.py b/code_to_image.py
index 0c36ba5..ed93be8 100644
--- a/code_to_image.py
+++ b/code_to_image.py
@@ -102,7 +102,7 @@ def draw_while_loop_back(condition: str, x: int, y: int, xsize: int, ysize: int)
 
 def NSD_save(filename: str):
     """Save the created file"""
-    img.save(filename + datei_endung ,".png")
+    img.save(filename + datei_endung ,"PNG")
 
 #x_offset , y_offset, x_size, y_size = draw_while_loop("lol", 0, 0, 100, 200)
 if __name__ == "__main__":
diff --git a/to_nassi.py b/to_nassi.py
index 6f17416..3bfd794 100644
--- a/to_nassi.py
+++ b/to_nassi.py
@@ -5,4 +5,8 @@ from Iinstruction import *
 def nassi(filepath:str, output_path: str):
     NSD = NassiShneidermanDiagram(False)
     NSD.load_from_file(filepath)
-    NSD.convert_to_image(output_path + "/Nassi-Shneider-Diagramm", 500)
\ No newline at end of file
+    NSD.convert_to_image(output_path + "/Nassi-Shneider-Diagramm", 500)
+
+if __name__ == "__main__":
+    """Debugging"""
+    nassi("res/input/input.java", "res/output")
\ No newline at end of file