Added comment to explain code
Minor changes