> Thought maybe there is a good way to do this.

Does that code work?
It's pretty readable to me, easy to understand what is going on.
There might be a better way, but if that works I see nothing wrong with it.