Update highlight.py