Python If Statement In One Line