Add comments to code

Comment code