if elif else statement in python single line