Updated some comments
Minor changes