Changed comment wording and made cosmetic changes to code