Address code review comments.

Auditors: markcc