Thanks for the suggestion. The code is open sourced on github. Feel free to do a pull request with your approach and I'll take a look at it.