Added comments
Minor edits