Update overlay.py