Yes, I partially used their approach which I found very smart. Added a few improvements which you can find on the GitHub.