Minor modifications to run/experiment.py