Updated run.py with the arguments