How to implement nested while loop in Python