Fixes to run_experiment script.