Updated comments to generate documentation