Updated test section and adjusted comments to align to format.