Update answer.py