Update wreak.py