I think you can skip this. You probably use it to check for converge, but it's not really needed here. A few extra iterations won't hurt that much, and the code should be simpler without this.