I doubt very much that the question asked was the real question. To that extent any answer is bogus and refinements, while usefull for teaching in a broader sense, are not actually addressing OP's real problem.

I consider that the code I posted as an answer to the question asked to be sub-optimum in many ways, but unless we can figure out what OP is trying to do, we probably can't do much better.