Jacobi and gauss seidel python