I would love to see this as well. If I have some time this weekend, I'll look into submitting a PR :)