Updated comment from code review