if statement in for loop in python