Add comments to the SGD code.