Update the comments

Add comments to the file