Gauss seidel with relaxation python