if else inside for loop in python