I'll try working it out with your method, but I may end up keeping the original function in the long run.

As much as I hate doing it the 'wrong' way, keeping the simplicity and readability may be a preferable option.