python if else pass one line