Python If Statement In Single Line