Update main.py

Updated for 48 lights