This change doesn't change the logic of the code. I just rewrote the code to make it better readable.