How to use range() with while loop