Implementing gauss seidel 2d method python