Updated comments
Minor changes