I think your change wouldn't be substantially different from that one. I'd be happy to code review / answer questions that come up.