Update zap.py