Corrected comments so they reflected the changes made