Gauss seidel method code python