Updated comments

added a few comments to code