while loop in python 3