Define while loop inside a for loop