The modified code is not mine to share - I made a promise that I wouldn't let it out in the wild and am standing by that promise - sorry.
The standard code is perfectly useable, just not as user friendly. It does exactly the same thing at the end of the day.