I started trying to answer this question a couple of times. The code is pretty straight-forward, and certainly fixing whatever is wrong should be pretty simple.

But I failed to come up with a solution because I don't know what the problem is. It's not that I don't understand how the code works. It's that I don't understand how it works differently from your needs and expectations, and that is because your needs and expectations haven't been expressed clearly and unambiguously enough.

Could you please follow-up by posting how the output this produces differs from the output you require? Maybe you could show the first 30 lines of correct output, and the first 30 lines of incorrect output. Or just explain better what you want to see.