Python One Liner If Statement