Update test_run.py