Minor comments update

Minor comments update