diff --git a/cage_controller.py b/cage_controller.py
index 716a160..8cd7eff 100644
--- a/cage_controller.py
+++ b/cage_controller.py
@@ -1,6 +1,6 @@
 import serial # Stuff for controlling the power supplies
 import time # Stuff for regulated sensor delays
-import smbus # Stuff for controlling temperature and magnetic sensors
+import smbus2 # Stuff for controlling temperature and magnetic sensors
 import utilities as utils # Stuff for debugging and/or general info
 from gpiozero import LED
 
diff --git a/command_line.py b/command_line.py
index acf826a..22f9c70 100755
--- a/command_line.py
+++ b/command_line.py
@@ -122,4 +122,9 @@ def poll_data(duration = 10.0, dt = 1.0):
         time_step.append(time_step[-1] + dt)
         # temp_array.append(temperature())
         mag_array.append(cage_controller.magnetometer())
-    return time_step, mag_array #temp_array, mag_array
\ No newline at end of file
+    return time_step, mag_array #temp_array, mag_array
+    
+if __name__ == "__main__":
+    interface()
+
+