Transposed wheels in machine str

src/py/slots/Slots.py

@@ -18,16 +18,29 @@ class SlotMachine():
 
     def __str__(self):
         string = ""
-        for wheel in self.wheels:
+        # Create an empty matrix with the sole purpose of transposing it
-              string += str(wheel) + "\n"
+        matrix = [[],[],[]]
+
+        for i, wheel in enumerate(self.wheels):
+            for icon in wheel.icons:
+                matrix[i].append(icon)
+                
+        # Use Zip function to aggregate unpacked matrix (*matrix unpacks it)
+        matrix = zip(*matrix)
+
+        for row in matrix:
+            for icon in row:
+                string += str(icon) + " "
+            string += "\n"
+            
         return string
 
 
 def main():
-    # machine = SlotMachine()
+    machine = SlotMachine()
-    # print(machine)
+    print(machine)
-    # machine.spin(randomness=1)
+    machine.spin()
-    # print(machine)
+    print(machine)
     pass
 
 if __name__ == "__main__":