Added extra comments to code