Python Single Line If