diff --git a/main.py b/main.py
index 6cf9c06..0d98f72 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 File Count 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()
+    else:
+        plt.show()
 
 
 if __name__ == '__main__':
diff --git a/media/example_output.png b/media/example_output.png
index 25d818e..593fb62 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..40bd4ab 100644
Binary files a/media/example_output2.png and b/media/example_output2.png differ