diff --git a/aliens.py b/aliens.py
index 97e83ca..97a9f0d 100644
--- a/aliens.py
+++ b/aliens.py
@@ -2,7 +2,7 @@
 def alien_table(index):
     return {
 
-        0: "Human",
+        0: "OFF",
         1: "Cannonbolt",
         2: "Ripjaws",
         3: "Diamondhead",
@@ -15,5 +15,27 @@ def alien_table(index):
         10: "Ghostfreak",
         11: "Wildvine",
         12: "Heatblast",
+        #TODO: Add more aliens
+    }.get(index, "ERROR")
 
-    }.get(index, "error")
\ No newline at end of file
+
+# screen_heatblast = PhotoImage(file="res/omnitrix.png")
+
+# def image_display(index):
+#     return {
+
+#         0: "res/omnitrix.png",
+#         1: "res/heatblast.png",
+#         2: "Ripjaws",
+#         3: "Diamondhead",
+#         4: "Stinkfly",
+#         5: "Fourarms",
+#         6: "Upgrade",
+#         7: "XLR8",
+#         8: "Grey Matter",
+#         9: "Wildmutt",
+#         10: "Ghostfreak",
+#         11: "Wildvine",
+#         12: "Heatblast",
+
+#     }.get(index, "res/omnitrix.png")
\ No newline at end of file
diff --git a/functions.py b/functions.py
index 5b5277a..a5376ba 100644
--- a/functions.py
+++ b/functions.py
@@ -1,4 +1,4 @@
-from aliens import alien_table
+from aliens import *
 
 counter= 0
 
@@ -19,3 +19,7 @@ def transform():
         print(alien_table(counter))
     elif counter > 0:
         print(alien_table(counter))
+#     change_image(counter)
+
+# def change_image(counter):
+#     omnitrix_screen = PhotoImage(file= image_display(counter))