changes to light colors fn to take light as loop

before the fn was hardcoded to three lights