Fixed comment above while loop