Added/modified comments to code