Add comments to the code