Python While Loop With Return