Update experiment.py