Update convergence_test.py