Added additional comments

Updated code comments to improve understanding