Add comment after code review