One Liner If Statement Python