Python If In Single Line