Fixed comment again and adjusted code to match the description.