Sounds like a good idea. I'll update the PR shortly.