diff --git a/docs/results.png b/docs/results.png
index 475caa4..83c2a09 100644
Binary files a/docs/results.png and b/docs/results.png differ
diff --git a/src/visuals/results.py b/src/visuals/results.py
index ac691b1..5f5adc6 100644
--- a/src/visuals/results.py
+++ b/src/visuals/results.py
@@ -1,6 +1,6 @@
 
 import os
-import time
+import matplotlib
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.patches import Ellipse