Update perf_test.py