Update audit.py