Conditional more pythonic