Gauss seidel implmentation python