python for loop with if else statement