diff --git a/christmastree.py b/christmastree.py index 881fd04..086cca7 100644 --- a/christmastree.py +++ b/christmastree.py @@ -30,4 +30,7 @@ def __init__(self, pwm=False, initial_value=False, pin_factory=None): _order=('baubles', 'star'), pin_factory=pin_factory ) + + self.red_leds = tuple(self.leds[i] for i in [1, 3, 4, 6, 8]) + self.green_leds = tuple(self.leds[i] for i in [0, 2, 5, 7])