Single Line Python If Statement