Update worker_demo.py