Python While Loop With Assignment