update per code review feedback