Update parallel_jacobi_working.py