Cool, I'll put it in the railtie. I'm having a little trouble getting some tests to work, so it may be a bit before I push the change to the PR.