One Liner Python If Statement