Refactor generated code.

Signed-off-by: John McCutchan <john@johnmccutchan.com>