One Liner If Else Python