Sounds good. I've added another commit to do just that.