Adding while loop section