Remove outdated comments and imports from run_experiment.py