Update survey.py