Minor comments update
Minor updates