Update per code review