update code according to the code review comment