Update lights.py