diff --git a/diagnostics/HubDiagnostics.py b/diagnostics/HubDiagnostics.py
index 9bdf436..ee5113f 100644
--- a/diagnostics/HubDiagnostics.py
+++ b/diagnostics/HubDiagnostics.py
@@ -1,13 +1,13 @@
 from pybricks.tools import wait, StopWatch
-from pybricks.parameters import Port
 from pybricks import version
 import OtherFunctions as debug
 import usys
 
-
+print("Pybricks version information:", version)
+print("MicroPython information:", usys.implementation)
+print("MicroPython version:", usys.version)
 class HubDiagnostics:
     def __init__(self, hub):
-        self.hub = hub
         self.port_map = {
             "A": Port.A,
             "B": Port.B,
@@ -16,20 +16,13 @@ class HubDiagnostics:
             "E": Port.E,
             "F": Port.F,
         }
-    def printAbout(self):
-        print("Pybricks version information:", version)
-        print("MicroPython information:", usys.implementation)
-        print("MicroPython version:", usys.version)
     def testLightSources(self, verbose):
         v = verbose
         self.hub.display.off()
         for x in range(5):
             for y in range(5):
                 debug.log(f"Turning on pixel at position {x}, {y}...", v)
-                self.hub.display.pixel(x, y, brightness=100)
+                display.pixel(x, y, brightness=100)
                 wait(100)
                 debug.log(f"Turning off pixel at position {x}, {y}...", v)
-                self.hub.display.pixel(x, y, brightness=0)
-    def printAll(self):
-        self.printAbout()
-        self.testLightSources(False)
+                display.pixel(x, y, brightness=0)
diff --git a/diagnostics/OtherFunctions.py b/diagnostics/OtherFunctions.py
index e5670f6..a7638a9 100644
--- a/diagnostics/OtherFunctions.py
+++ b/diagnostics/OtherFunctions.py
@@ -1,3 +1,4 @@
-def log(string, verbose):
+verbose = True
+def log(string):
     if(verbose):
         print("[LOG (verbose)]", string)