Update experiment.py

Trying to tidy up code to find the error in the AuditFailure line.