Single Line If Statement Python