python single line if elif else statement