diff --git a/src/py/slots/Slots.py b/src/py/slots/Slots.py
index 9adb1e1..c4279e8 100644
--- a/src/py/slots/Slots.py
+++ b/src/py/slots/Slots.py
@@ -1,6 +1,9 @@
 from Wheel import Wheel
 
 class SlotMachine():
-    def __init__(self):
+    def __init__(self, screen_w=3, screen_h=3):
         jackpot = 0
-        wheel = Wheel()
\ No newline at end of file
+        wheels=[]
+        for i in range(screen_w):
+            wheel = Wheel(size=screen_h)
+            wheels.append(wheel)
\ No newline at end of file