CHanges to run_test.py