Ok, did a quick try at this while I was in the plane and after some cleaning up sent a pull request.
Here is it in action: