Modification to perf_testing.py