Update locust.py