python if else syntax one line