On the contrary, your confusion shows that my suggestion is good ;-) It's easier to read if each branch does the same thing.