Update track_demo.py