Fixed the for loop in run.py