Added comment to code