diff --git a/.gitignore b/.gitignore
index f345153..7033066 100644
--- a/.gitignore
+++ b/.gitignore
@@ -129,4 +129,5 @@ dmypy.json
 # Pyre type checker
 .pyre/
 
-.idea/
\ No newline at end of file
+.idea/
+data/database.db
diff --git a/speedtester.py b/speedtester.py
index 31ee8ec..88082ca 100644
--- a/speedtester.py
+++ b/speedtester.py
@@ -28,6 +28,7 @@ def generate_plot_image(dates, downloads, uploads):
     import matplotlib.pyplot as plt
 
     dates = matplotlib.dates.date2num(dates)
+    fig = plt.figure(figsize=(12,3))
     plt.plot_date(dates, downloads, fmt="b-")
     plt.ylabel('Download Speed Mbps')
     plt.tight_layout()
@@ -49,6 +50,10 @@ if __name__ == "__main__":
     This script will run a few speed tests, calculate average upload and download speeds and record them into database.
     Once finished it will also generate an image with graph plotted.
     '''
+
+    # generate_plot_image(*gather_data())
+    # import sys
+    # sys.exit()
     from random import uniform
     try:
         import speedtest