Corrected comment code to fit function behavior for week 1.