I tweaked it just a little to make it fit in a bit better with the rest of the code and fix a few lint warnings.

Thanks again for your PR @zaquestion!