Update attack.py