diff --git a/main.py b/main.py
index 6cf9c06..876ff98 100644
--- a/main.py
+++ b/main.py
@@ -122,15 +122,16 @@ def main(args):
 
     plt.xlabel('Date', fontsize=14)
     plt.ylabel('Count', fontsize=14)
-    plt.title('Test File Count by extension in Instrumentation Directory', fontsize=16)
+    plt.title('Test Files by Language in Instrumentation Directory', fontsize=16)
     plt.xticks(rotation=45)
 
     plt.legend()
     plt.tight_layout()
 
-    if len(args.output) > 0:
+    if args.output is not None:
         plt.savefig(args.output)
-    plt.draw()
+
+    plt.show()
 
 
 if __name__ == '__main__':
diff --git a/media/example_output.png b/media/example_output.png
index 25d818e..9b6d011 100644
Binary files a/media/example_output.png and b/media/example_output.png differ
diff --git a/media/example_output2.png b/media/example_output2.png
index 954ee51..2e46598 100644
Binary files a/media/example_output2.png and b/media/example_output2.png differ