update segment.py